This repo contains the code for conference paper titled Flower Species Recognition System using Convolutional Neural Networks and Transfer Learning, by I.Gogul and V.Sathiesh Kumar, Proceedings of ICSCN-2017 conference, IEEE Xplore Digital Library.
Update (16/12/2017): Included two new deep neural net models namely InceptionResNetv2
and MobileNet
.
sudo pip install theano
or sudo pip install tensorflow
sudo pip install keras
sudo pip install numpy
sudo pip install matplotlib
and you also need to do this sudo apt-get install python-dev
sudo pip install seaborn
sudo pip install h5py
sudo pip install scikit-learn
MIT License
python organize_flowers17.py
python extract_features.py
python train.py
The below tables shows the accuracies obtained for every Deep Neural Net model used to extract features from FLOWERS17 dataset using different parameter settings.
Result-1
Model | Rank-1 accuracy | Rank-5 accuracy |
---|---|---|
Xception | 97.06% | 99.26% |
Inception-v3 | 96.32% | 99.26% |
VGG16 | 85.29% | 98.53% |
VGG19 | 88.24% | 99.26% |
ResNet50 | 56.62% | 90.44% |
MobileNet | 98.53% | 100.00% |
Inception ResNetV2 |
91.91% | 98.53% |
Result-2
Model | Rank-1 accuracy | Rank-5 accuracy |
---|---|---|
Xception | 93.38% | 99.75% |
Inception-v3 | 96.81% | 99.51% |
VGG16 | 88.24% | 99.02% |
VGG19 | 88.73% | 98.77% |
ResNet50 | 59.80% | 86.52% |
MobileNet | 96.32% | 99.75% |
Inception ResNetV2 |
88.48% | 99.51% |