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 -2
View File
@@ -24,7 +24,7 @@ test_gen = ImageDataGenerator().flow_from_directory(
batch_size=batch_size,
shuffle=False
)
#
single_gen = ImageDataGenerator().flow_from_directory(
'./single_image_test_set',
target_size=(input_shape[0], input_shape[1]),
@@ -32,7 +32,6 @@ single_gen = ImageDataGenerator().flow_from_directory(
shuffle=False
)
for file in glob("./models/keras/*.hdf5"):
print(file)
if file in metrics_df.values: