Maritime Vessel Recognition
The models shown below were trained using an in-distribution dataset.
Models
| Checkpoint Name | Loss Function | Balanced Training Data | Training Module |
|---|---|---|---|
| vit-b-16.pth | Cross Entropy | - | TorchVision |
| vit-b-16_BTD.pth | Cross Entropy | 200k per category | TorchVision |
| vit-b-16_LAL.pth | Logit Adjusted Loss | - | TorchVision |
| vit-b-16_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TorchVision |
| vit-l-16.pth | Cross Entropy | - | TorchVision |
| vit-l-16_BTD.pth | Cross Entropy | 200k per category | TorchVision |
| vit-l-16_LAL.pth | Logit Adjusted Loss | - | TorchVision |
| vit-l-16_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TorchVision |
| vit-h-14.pth | Cross Entropy | - | TorchVision |
| vit-h-14_BTD.pth | Cross Entropy | 200k per category | TorchVision |
| vit-h-14_LAL.pth | Logit Adjusted Loss | - | TorchVision |
| vit-h-14_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TorchVision |
| vit-b-16-timm.pth | Cross Entropy | - | TIMM |
| vit-b-16-timm_BTD.pth | Cross Entropy | 200k per category | TIMM |
| vit-b-16-timm_LAL.pth | Logit Adjusted Loss | - | TIMM |
| vit-b-16-timm_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TIMM |
| vit-l-16-timm.pth | Cross Entropy | - | TIMM |
| vit-l-16-timm_BTD.pth | Cross Entropy | 200k per category | TIMM |
| vit-l-16-timm_LAL.pth | Logit Adjusted Loss | - | TIMM |
| vit-l-16-timm_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TIMM |
| vit-h-14-timm.pth | Cross Entropy | - | TIMM |
| vit-h-14-timm_BTD.pth | Cross Entropy | 200k per category | TIMM |
| vit-h-14-timm_LAL.pth | Logit Adjusted Loss | - | TIMM |
| vit-h-14-timm_BTDandLAL.pth | Logit Adjusted Loss | 200k per category | TIMM |