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

Downloads last month: 4

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Examples

Image Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/vessel_ship_types_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1643)

this model