kbarnard commited on
Commit
8cb9e19
1 Parent(s): a077dd2

Write first cut of model card

Browse files
Files changed (1) hide show
  1. README.md +42 -0
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
+ ```