Image classification of for Kaggle cats and dogs challenge based on tutorial on Keras web blog.
A ConvNet model following the tutorial at the blog. The model parameters were further saved to load the model and perform sentiment analysis.
Data source: Kaggle Cats vs Dogs.
Future prospects:
- Improve model with different algorithms.
- Try more pre-trained image classification models.
- Root: Main folder containing Readme and scripts.
- data: Data folder containing:
a) train: Training data.
b) validation: Test/Validation data.
c) test: Test data. - figures: All generated figures from the scripts.
- models: Trained models stored in this folder.
utilities.py: Contains useful functions.
ImageClassification.py: ImageClassification file.
CheckImages.py: Image augmentation example.
=======================================================
utilities.py:
This script is required for all other scripts.
Useful common functions.
=======================================================
ImageClassification.py:
Image classification using ConvNet neural networks.
With simple network validation accuracy are 75%.
Large fluctuations in accuracy history.
Loss function and Accuracies are plotted in Figures:
Accuracy.png
Loss.png