Dataset size: 4839 images
I changed the model name in Jupyter Notebook to popular image classification models.
List of models: https://keras.io/api/applications/
MobileNetV2
- Between 37 and 48 seconds per epoch
- Reached 100% accuracy on the first epoch
InceptionResNetV2
- This model link didn’t work:
So, I tested a different link for the same model and it worked
- Between 95 and 96 seconds per epoch
- Reached 100% accuracy after 1 epoch
NASNetMobile
- Between 74 and 84 seconds per epoch
- Reached 100% accuracy on the first epoch
EfficientNet_V2
- Between 51 and 56 seconds per epoch
- Reached 100% accuracy on the first epoch
bit
- Between 288 and 319 seconds per epoch
- Reached 100% on the first epoch
Ultimately, the MobileNet V2 model I originally used was the most time-efficient at 37 to 48 seconds for each epoch. All of the models reached 100% accuracy on the first epoch.
