Dataset:
This deep learning project to classify 6 types of wild animals using CNN.
Classes:
1.Cheetah 2.Fox 3.Hyena 4.Lion 5.Tiger 6.Wolf
Data colletion:
Wild animals image dataset collected from kaggle.Image_size(224,224) and each classes had 100 images.
Features:
Custom convolutional neural network with 5 layers(1-input,3-hidden and 1-output layers),also using data augmentation to improve generalization.
Tools:
language Python , Tensorflow and keras, Jupyter notebook.
Save:
Model save in h5(Hierarchical file)
Model got at 85% accuracy of training by after epoch 35/50.
Validation accuracy:
Reach 65% (Improvement need)
Model save in h5 (Hirarchical data format5) for prediction.
In this project image classification for CNN using deeplearning in python.UI for upload file then classify it Iron man or ultron.More image dataset for enhance prediction will be achive.