feat: adding resnet and formatting updates

This commit is contained in:
Lucas Oskorep
2023-04-06 00:37:46 -04:00
parent ce5939d8a9
commit dc427837f6
12 changed files with 87 additions and 37 deletions
-1
View File
@@ -19,7 +19,6 @@ for index, row in df2.iterrows():
incorrect = df[df["prediction"]!= df["true_val"]]
total_same_fam = 0
# TODO: Add in support for figuring out if the pokemon are related/evolutions of one another
for index, row in incorrect.iterrows():
img = mpimg.imread("./SingleImageTestSet/" + row['fname'])
imgplot = plt.imshow(img)