Update README.md
Browse files
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 |
-

|
| 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 |
+

|
|
|