Update README.md
Browse files
README.md
CHANGED
|
@@ -10,4 +10,57 @@ pipeline_tag: object-detection
|
|
| 10 |
tags:
|
| 11 |
- biology
|
| 12 |
- climate
|
| 13 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
tags:
|
| 11 |
- biology
|
| 12 |
- climate
|
| 13 |
+
---
|
| 14 |
+
# Model Card for Model ID
|
| 15 |
+
|
| 16 |
+
This model is designed for object detection in marine environments, specifically targeting benthic megafauna in ROV transects.
|
| 17 |
+
|
| 18 |
+
## Model Details
|
| 19 |
+
|
| 20 |
+
### Training Data
|
| 21 |
+
|
| 22 |
+
The model is trained on the OceanCV/EcotypeTransect dataset, which contains annotated images of marine fauna and geological features from various methane seep and hydrothermal vent sites along the West Coast.
|
| 23 |
+
|
| 24 |
+
### Training Procedure
|
| 25 |
+
|
| 26 |
+
Preprocessing
|
| 27 |
+
|
| 28 |
+
Images were resized to 1024x1024 pixels
|
| 29 |
+
|
| 30 |
+
Data augmentation techniques included random flipping, rotation, and contrast adjustments
|
| 31 |
+
|
| 32 |
+
Training Hyperparameters
|
| 33 |
+
|
| 34 |
+
Training regime: Mixed precision (fp16)
|
| 35 |
+
Epochs: 200
|
| 36 |
+
Batch size: Auto (-1)
|
| 37 |
+
Optimizer: Auto
|
| 38 |
+
Learning rate: 0.01
|
| 39 |
+
Momentum: 0.937
|
| 40 |
+
Weight decay: 0.0005
|
| 41 |
+
|
| 42 |
+
|
| 43 |
+
Model size: 450 MB
|
| 44 |
+
|
| 45 |
+
### Evaluation
|
| 46 |
+
|
| 47 |
+
Testing Data, Factors & Metrics
|
| 48 |
+
|
| 49 |
+
Testing Data
|
| 50 |
+
|
| 51 |
+
The testing dataset consists of near methane seep imagery from a singular 38 ROV surveys conducted in the Northeast Pacific.
|
| 52 |
+
|
| 53 |
+
### Metrics
|
| 54 |
+
|
| 55 |
+
mAP50: 78.0%
|
| 56 |
+
mAP50-95: 56.7%
|
| 57 |
+
Precision: 72.7%
|
| 58 |
+
Recall: 75.1%
|
| 59 |
+
|
| 60 |
+
- **Developed by:** Atticus Carter
|
| 61 |
+
|
| 62 |
+
- **Model type:** YOLOV11
|
| 63 |
+
|
| 64 |
+
- **License:** MIT
|
| 65 |
+
|
| 66 |
+
- **Finetuned from model:** FathomNet/MBARI-315k-yolov8
|