Downloads · 30 days
628
27% of all-time downloads
vessel_ship_types_image_detection
dima806/vessel_ship_types_image_detection
vessel_ship_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 vessel/ship type based on image with about 99% accuracy.
All-time downloads
2.3K
Public
Parameters
85.8M
3.1 GB on disk
Likes
3
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 60%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Dec 2, 2023
- Updated
- Oct 19, 2024
- SHA
- 885ca14f
Base models
Model card
From the Hugging Face model README
Returns vessel/ship type based on image with about 99% accuracy.
See https://www.kaggle.com/code/dima806/vessel-ship-type-detection for more details.
Classification report:
precision recall f1-score support
Cargo 0.9927 0.9623 0.9772 424
Carrier 0.9976 1.0000 0.9988 424
Cruise 1.0000 1.0000 1.0000 424
Military 0.9976 0.9976 0.9976 424
Tankers 0.9679 0.9953 0.9814 424
accuracy 0.9910 2120
macro avg 0.9912 0.9910 0.9910 2120
weighted avg 0.9912 0.9910 0.9910 2120
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Dec 2, 2023
- Updated
- Oct 19, 2024