From 63c6ae62b3624a2a8a76bbdf7fe86d15671ae501 Mon Sep 17 00:00:00 2001 From: Lucas Oskorep Date: Sun, 14 Apr 2019 15:19:23 -0500 Subject: [PATCH] Updating the input image sizes to have a higher resolution. Currently unclean data is reaching 70% accuracy. --- TrainingModelKeras.py | 2 +- TransferLearningKeras.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/TrainingModelKeras.py b/TrainingModelKeras.py index b6fde87..14c4799 100755 --- a/TrainingModelKeras.py +++ b/TrainingModelKeras.py @@ -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" diff --git a/TransferLearningKeras.py b/TransferLearningKeras.py index 6a5c46c..855b7aa 100755 --- a/TransferLearningKeras.py +++ b/TransferLearningKeras.py @@ -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"