Update README.md
Browse files
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 |
+
```
|