Instructions to use FishingROV/classifier_swinv2b_256 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use FishingROV/classifier_swinv2b_256 with timm:
import timm model = timm.create_model("hf_hub:FishingROV/classifier_swinv2b_256", pretrained=True) - Notebooks
- Google Colab
- Kaggle
File size: 2,178 Bytes
792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 792b4dc c998083 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | ---
license: cc-by-4.0
library_name: timm
pipeline_tag: image-classification
tags:
- swinv2
- image-classification
- underwater
- marine-biology
- scallop
- rov
- fisheries
---
# FishingROV - SwinV2-B 256 classifier (crops)
Zoo ID: `cls-classifier_swinv2b_256` - canonical weights: `best.pt`
SwinV2-B classifier for 5-way scallop taxonomy on square crops. Used in the
3090-side detector -> crop -> classifier pipeline (paired with the YOLO teacher
for ROI generation). This is a classifier only; it does not perform detection.
## Dataset provenance (station-disjoint)
This model is trained on `DS-CLS224` (`classifier_data`) generated from the
public Zenodo dataset:
- Train split: Zenodo **Training files** stations
- Val split: Zenodo **Test files** stations (station-disjoint)
Crops are square, centered on human boxes, padded if needed, resized to 224px.
Negatives are sampled away from GT boxes. No augmentation.
## Metrics (same-crop validation)
All metrics below are from `classifier_data/val` (station-disjoint Test files).
| Metric | Value |
| --- | --- |
| Macro precision | 0.700 |
| Macro recall | 0.654 |
| Macro F1 | 0.661 |
| Accuracy | 0.966 |
Per-class metrics (from `class_eval_best.json`):
| Class | Precision | Recall | F1 | Support |
| --- | --- | --- | --- | --- |
| dead | 0.464 | 0.642 | 0.539 | 81 |
| king | 0.391 | 0.237 | 0.295 | 76 |
| not_a_scallop | 0.991 | 0.996 | 0.993 | 5781 |
| queen | 0.818 | 0.899 | 0.857 | 296 |
| recessed | 0.837 | 0.497 | 0.623 | 145 |
## Intended use & limitations
- Intended for crop classification, not end-to-end detection.
- Trained only on the St Andrews survey distribution; generalization is unknown.
- King class remains the weakest; expect confusion in borderline cases.
## Files
- `best.pt` - best checkpoint (SwinV2-B 256)
- `last.pt` - final checkpoint
- `class_eval_best.json` - macro + per-class metrics
- `classes.json`, `history.json`
## Attribution & License
This model is a derivative work based on the **University of St Andrews King
Scallop dataset**.
- Original DOI: https://doi.org/10.5281/zenodo.10156830
Released under **CC-BY 4.0** with attribution to the original authors.
|