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 |
+
```
|