karanwxliaa / Underwater-Trash-Detection

Underwater Trash Detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Underwater Trash Detection (custom dataset)

Due to highly variying domain features of different underwater enviornment, the publically available datasets alone are not the best fit to train a deep learning algorithm to predict trash. Therefore we propose a cumulatuve, self-annonated dataset that provides a good foundation for training models to detect and classify trash underwater, and also provide benchmarks for the same. This repository aims to generate code that detects trash, classifies it into plastic / trash / underwater debris against other factors like fish / flora & fauna and input rover images.

Aiming at the problem of insufficient storage space and limited computing ability of underwater mobile devices, an underwater garbage detection algorithm is proposed.

Yolov8x

image image

Yolov8n (nano)

image image

Yolov8 (small)

since YoloV8 is quite new and still under development it had a lot of bugs and didnt give the best outputs. image

Yolo V6s

Gave more accurate detections from v5 and lower false positives but better accuracy at the cost of lower detections. ezgif-2-5d833929c0 ezgif-2-86293c1b98 (1)

RESULTS

image

YOLOv5s

Higer detections, higher false negatives relative to v6 but more detections. ezgif-2-16e87df07e ezgif-2-67e77099d4

RESULTS

image

RCNN

image