Karl-Erik commited on
Commit
897e02d
·
1 Parent(s): 91da7ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -9,15 +9,15 @@ metrics:
9
  - accuracy
10
  ---
11
  # Introduction
12
- Hello and welcome to the Estonian Bird Classifier model page! This model was 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
 
18
- 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.
19
  On the test set, the model achieved 74% accuracy, 89% Top-3 accuracy and 91% Top-5 accuracy with a loss of 1.119.
20
- 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.
21
 
22
  # How to use
23
  **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.
 
9
  - accuracy
10
  ---
11
  # Introduction
12
+ Hello, and welcome to the Estonian Bird Classifier model page! This model was 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 complete list of birds that it can classify can be found in the label map provided with the model.
17
 
18
+ 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.
19
  On the test set, the model achieved 74% accuracy, 89% Top-3 accuracy and 91% Top-5 accuracy with a loss of 1.119.
20
+ 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.
21
 
22
  # How to use
23
  **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.