lucasoskorep/tensordex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Lucas Oskorep
2019-04-14 15:19:23 -05:00
parent fce361ddcb
commit 63c6ae62b3
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TrainingModelKeras.py
+1 -1
View File
@@ -9,7 +9,7 @@ from time import time
from PIL import ImageFile from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True ImageFile.LOAD_TRUNCATED_IMAGES = True
input_shape = (224, 224, 3) input_shape = (299, 299, 3)
batch_size = 32 batch_size = 32
model_name = "MobileNetV2FullDatasetNoTransfer" model_name = "MobileNetV2FullDatasetNoTransfer"
TransferLearningKeras.py
+2 -1
View File
@@ -8,7 +8,8 @@ from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True ImageFile.LOAD_TRUNCATED_IMAGES = True
input_shape = (224, 224, 3) input_shape = (299, 299, 3)
batch_size = 60 batch_size = 60
model_name = "MobileNetV2FullDataset" model_name = "MobileNetV2FullDataset"