Updating the input image sizes to have a higher resolution. Currently unclean data is reaching 70% accuracy.

2019-04-14 15:19:23 -05:00
parent fce361ddcb
commit 63c6ae62b3
2 changed files with 3 additions and 2 deletions
@@ -9,7 +9,7 @@ from time import time
 from PIL import ImageFile
 ImageFile.LOAD_TRUNCATED_IMAGES = True

-input_shape = (224, 224, 3)
+input_shape = (299, 299, 3)
 batch_size = 32
 model_name = "MobileNetV2FullDatasetNoTransfer"

@@ -8,7 +8,8 @@ from PIL import ImageFile

 ImageFile.LOAD_TRUNCATED_IMAGES = True

-input_shape = (224, 224, 3)
+input_shape = (299, 299, 3)
+
 batch_size = 60
 model_name = "MobileNetV2FullDataset"