Write first cut of model card
Browse files
README.md
CHANGED
@@ -1,3 +1,45 @@
|
|
1 |
---
|
2 |
license: cc-by-4.0
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: cc-by-4.0
|
3 |
+
tags:
|
4 |
+
- ocean
|
5 |
+
- object-detection
|
6 |
---
|
7 |
+
|
8 |
+
# FathomNet Vulnerable Marine Ecosystems (VME) Detector
|
9 |
+
|
10 |
+
## Model Details
|
11 |
+
|
12 |
+
- Trained by researchers at the [Monterey Bay Aquarium Research Institute](https://www.mbari.org/) (MBARI).
|
13 |
+
- Ultralytics [YOLOv8x](https://github.com/ultralytics/ultralytics)
|
14 |
+
- Object detection model
|
15 |
+
- Fine-tuned to detect 4 high-level classes of benthic animals from deep-sea imagery specifically identified as indicators of vulnerable marine ecosystems
|
16 |
+
- These VME categories include *corals*, *crinoids*, *sponges*, and *fishes*
|
17 |
+
- [Baco et al. 2023](https://peerj.com/articles/16024) (Table 2) was used to determine classes that were useful for detecting VME's, however we added fishes as an additionalclass due to the undeniable fact that VMEs and fishery management often overlap
|
18 |
+
|
19 |
+
## Intended Use
|
20 |
+
|
21 |
+
- Post-process video and images collected by marine researchers to determine presence of VME indicator species
|
22 |
+
|
23 |
+
## Factors
|
24 |
+
|
25 |
+
- Distribution shifts related to sampling platform, camera parameters, illumination, and deployment environment are expected to impact model performance
|
26 |
+
- Evaluation was performed on an IID subset of available training data as well as out-of-distribution data
|
27 |
+
|
28 |
+
## Metrics
|
29 |
+
|
30 |
+
- [Normalized confusion matrix](plots/confusion_matrix_normalized.png), [precision-recall curve](plots/PR_curve.png), and [F1-confidence curve](plots/F1_curve.png) were evaluated at test time
|
31 |
+
- [email protected] = 0.713
|
32 |
+
|
33 |
+
## Training and Evaluation Data
|
34 |
+
|
35 |
+
- Publicly-available data on [FathomNet](https://fathomnet.org/)
|
36 |
+
- TODO: Add specific class to concept mapping used to query FathomNet
|
37 |
+
|
38 |
+
## Deployment
|
39 |
+
|
40 |
+
1. Clone this repository
|
41 |
+
2. In an environment with the [`ultralytics` Python package](https://github.com/ultralytics/ultralytics) installed, run:
|
42 |
+
|
43 |
+
```bash
|
44 |
+
yolo predict model=best.pt
|
45 |
+
```
|