kbarnard commited on
Commit
94e7c92
1 Parent(s): ef3d407

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
+ - object-localization
7
+ - single-class
8
  ---
9
+
10
+ # FathomNet Megalodon Detector
11
+
12
+ ## Model Details
13
+
14
+ - Trained by researchers at the [Monterey Bay Aquarium Research Institute](https://www.mbari.org/) (MBARI).
15
+ - Ultralytics [YOLOv8x](https://github.com/ultralytics/ultralytics)
16
+ - Object detection model
17
+ - Fine-tuned to detect 1 class, called 'object', using all FathomNet localizations
18
+
19
+ ## Intended Use
20
+
21
+ - Post-process video and images collected by marine researchers
22
+ - Can be used to build a localized set of training images, when neither training data nor a model exists for the imagery being analyzed
23
+
24
+ ## Factors
25
+
26
+ - Distribution shifts related to sampling platform, camera parameters, illumination, and deployment environment are expected to impact model performance
27
+ - Evaluation was performed on an IID subset of available training data as well as out-of-distribution data
28
+
29
+ ## Metrics
30
+
31
+ - [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
32
+ - mAP@0.5 = 0.782
33
+
34
+ ## Training and Evaluation Data
35
+
36
+ - All publicly-available data on [FathomNet](https://fathomnet.org/)
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
+ ```