Downloads · 30 days
16
3% of all-time downloads
10_ship_types_image_detection
dima806/10_ship_types_image_detection
10_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 ship type given an image with about 99.6% accuracy.
All-time downloads
609
Public
Parameters
85.8M
4.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
- Oct 31, 2023
- Updated
- Oct 19, 2024
- SHA
- 85a457c4
Base models
Model card
From the Hugging Face model README
Returns ship type given an image with about 99.6% accuracy.
See https://www.kaggle.com/code/dima806/ship-type-detection-vit for more details.
Classification report:
precision recall f1-score support
Bulkers 0.9927 1.0000 0.9963 409
Recreational 0.9902 0.9927 0.9915 409
Sailboat 0.9975 0.9853 0.9914 409
DDG 0.9976 1.0000 0.9988 409
Container Ship 1.0000 0.9951 0.9975 409
Tug 0.9951 0.9927 0.9939 410
Aircraft Carrier 1.0000 0.9976 0.9988 409
Cruise 1.0000 1.0000 1.0000 409
Submarine 0.9927 1.0000 0.9964 410
Car Carrier 0.9951 0.9976 0.9963 409
accuracy 0.9961 4092
macro avg 0.9961 0.9961 0.9961 4092
weighted avg 0.9961 0.9961 0.9961 4092
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Oct 31, 2023
- Updated
- Oct 19, 2024