Karl-Erik commited on
Commit
9396e8c
·
1 Parent(s): 52fc1b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -1
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=";")