SoulPerforms commited on
Commit
f301e11
·
verified ·
1 Parent(s): 35a7dfc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -5,6 +5,10 @@ tags:
5
  metrics:
6
  - accuracy
7
  pipeline_tag: image-classification
 
 
 
 
8
  ---
9
  Butterfly image classification model that use pre-trained cnn model resnet18 and fine-tuned the last fully connected layer to classify 75 categories of butterfly species.
10
 
@@ -28,5 +32,4 @@ Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
28
  9. model_for_predict.eval())
29
 
30
 
31
- ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/65bdde53259bc6caeb094e85/b__Bo6f-R0el__5yqusRc.jpeg)
32
-
 
5
  metrics:
6
  - accuracy
7
  pipeline_tag: image-classification
8
+ datasets:
9
+ - huggan/inat_butterflies_top10k
10
+ language:
11
+ - en
12
  ---
13
  Butterfly image classification model that use pre-trained cnn model resnet18 and fine-tuned the last fully connected layer to classify 75 categories of butterfly species.
14
 
 
32
  9. model_for_predict.eval())
33
 
34
 
35
+ ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/65bdde53259bc6caeb094e85/b__Bo6f-R0el__5yqusRc.jpeg)