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=";")
|