Downloads · 30 days
12
2% of all-time downloads
diamond_types_image_detection
dima806/diamond_types_image_detection
diamond_types_image_detection is a image classification model from dima806. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Returns diamond type given an image with about 99% accuracy.
All-time downloads
540
Public
Parameters
85.8M
3.1 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 55%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Sep 10, 2023
- Updated
- Oct 19, 2024
- SHA
- 073498a0
Base models
Model card
From the Hugging Face model README
Returns diamond type given an image with about 99% accuracy.
See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.
Classification report:
precision recall f1-score support
pear 1.0000 0.9956 0.9978 2280
cushion 0.9418 1.0000 0.9700 2280
princess 0.9996 0.9890 0.9943 2279
round 0.9964 0.9781 0.9872 2279
marquise 0.9987 0.9882 0.9934 2279
oval 0.9996 0.9904 0.9949 2280
emerald 1.0000 0.9956 0.9978 2279
heart 0.9987 0.9943 0.9965 2280
accuracy 0.9914 18236
macro avg 0.9918 0.9914 0.9915 18236
weighted avg 0.9918 0.9914 0.9915 18236
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Sep 10, 2023
- Updated
- Oct 19, 2024