Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,28 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- et
|
| 5 |
+
- en
|
| 6 |
+
tags:
|
| 7 |
+
- image classifier
|
| 8 |
+
metrics:
|
| 9 |
+
- accuracy
|
| 10 |
+
---
|
| 11 |
+
# Introduction
|
| 12 |
+
Hello and welcome to the Estonian Bird Classifier model page! This model is created by Karl-Erik Kanal as a part of his Bachelor's thesis and can recognise 50 common Estonian bird species.
|
| 13 |
+
|
| 14 |
+
# About the model
|
| 15 |
+
The model estonian_birds_classifier is a pretrained InceptionV3 model on imagenet weights that has been trained using transfer learning to recognise 50 bird species that can be found in estonia.
|
| 16 |
+
The full list of birds that it can classify can be found in the label map provided with the model.
|
| 17 |
+
The model was trained and tested using a custom made dataset for this model, with 5926 images in the training set and 1064 images in the test set.
|
| 18 |
+
On the test set, the model achieved 74% accuracy, 89% Top-3 accuracy and 91% Top-5 accuracy with a loss of 1.119.
|
| 19 |
+
In comparison, the [Google AIY bird classifier](https://tfhub.dev/google/aiy/vision/classifier/birds_V1/1) that can recognise 45 species of the 50, achieved 71% accuracy on the test set with the 5 species taken out while this model achieved 75% accuracy with the same 45 species.
|
| 20 |
+
|
| 21 |
+
# How to use
|
| 22 |
+
**The model requires the images to be resized to 150 x 150 and normalized before predicting.** You can use the ImageDataGenerator class from keras.preprocessing.image to achieve this.
|
| 23 |
+
The images should be well-cropped to achieve the best results.
|
| 24 |
+
You can load in the model like you would load in other Keras type models by simply using the load_model function.
|
| 25 |
+
You can load the labels into a dataframe using pandas read_csv. The labels of the birds are provided both in Estonian and in Latin.
|
| 26 |
+
|
| 27 |
+
model = keras.models.load_model('estonian_birds_classifier.h5')
|
| 28 |
+
labels = pd.read_csv('ebc_labelmap.csv', sep=";")
|