kbarnard commited on
Commit
23a1c17
1 Parent(s): f1782f4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md CHANGED
@@ -1,3 +1,65 @@
1
  ---
2
  license: cc-by-4.0
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-4.0
3
+ tags:
4
+ - ocean
5
+ - object-detection
6
+ - trash
7
  ---
8
+
9
+ # Trash Detector
10
+
11
+
12
+ ## Model Details
13
+
14
+ - Trained by researchers at the Monterey Bay Aquarium Research Institute (MBARI).
15
+ - Ultralytics YOLOv8x
16
+ - Object detection model
17
+ - Classes included in this detection model:
18
+ - trash
19
+ - eel
20
+ - rov
21
+ - starfish
22
+ - fish
23
+ - crab
24
+ - plant
25
+ - animal_misc
26
+ - shells
27
+ - bird
28
+ - shark
29
+ - jellyfish
30
+ - ray
31
+
32
+ ## Intended Use
33
+
34
+ - Post-process video and images collected by marine researchers
35
+ - This model should do a reasonable job detecting marine debris in a variety of habitats, depths, and lighting conditions.
36
+ - Can be used to build a localized set of training images, when neither training data nor a model exists for the imagery being analyzed.
37
+
38
+ ## Factors
39
+
40
+ - Distribution shifts related to sampling platform, camera parameters, illumination, and deployment environment are expected to impact model performance
41
+
42
+ ## Metrics
43
+
44
+ TODO
45
+
46
+ ## Training and Evaluation Data
47
+
48
+ - Fine-tuned to detect 13 classes using training data combined from the following sources:
49
+ 1. MBARI/FathomNet
50
+ 2. trash-can: https://conservancy.umn.edu/handle/11299/214865
51
+ 3. deep plastic: https://github.com/gautamtata/DeepPlastic
52
+ 4. taco-dataset: https://tacodataset.org/
53
+ 5. ocean agency image bank: https://www.theoceanagency.org/search-result?s=trash
54
+ 6. Trash-ICRA19: https://conservancy.umn.edu/handle/11299/214366
55
+ 7. roboflow aquarium dataset
56
+ 8. roboflow Underwater Trash Detection.v5-dataset_v3
57
+ - A compiled list of trash training data sets is here: https://github.com/AgaMiko/waste-datasets-review
58
+
59
+ ## Deployment
60
+
61
+ 1. Clone this repository
62
+ 2. In an environment with the ultralytics Python package installed, run:
63
+ ```bash
64
+ yolo predict model=trash_mbari_09072023_640imgsz_50epochs_yolov8.pt
65
+ ```