SoulPerforms
/

Butterfly_image_classification_resnet18

Image Classification

Model card Files Files and versions

SoulPerforms commited on Feb 25, 2024

Commit

25bbb04

·

verified ·

1 Parent(s): 4277e7d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -5,7 +5,7 @@ tags:
 metrics:
 - accuracy
 pipeline_tag: image-classification
-library_name: bertopic
 ---
 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.
 The model used the best checkpoint with 90% test accuracy. The model constructed on Pytorch environment.

 metrics:
 - accuracy
 pipeline_tag: image-classification
+library_name: flair
 ---
 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.
 The model used the best checkpoint with 90% test accuracy. The model constructed on Pytorch environment.