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"