We can't find the internet
Attempting to reconnect
Something went wrong!
Hang in there while we get back on track
Efficient Object Detection of Marine Debris using Pruned YOLO Model
Summary
Researchers applied a pruned YOLOv4 object detection model to identify marine debris, including microplastic-contaminated items, in underwater images. Channel pruning reduced the model size by 90% while maintaining detection accuracy, making it suitable for deployment on autonomous underwater vehicles. The study demonstrates that efficient machine learning models can support real-time marine debris detection and cleanup efforts.
Marine debris poses significant harm to marine life due to substances like microplastics, polychlorinated biphenyls, and pesticides, which damage habitats and poison organisms. Human-based solutions, such as diving, are increasingly ineffective in addressing this issue. Autonomous underwater vehicles (AUVs) are being developed for efficient sea garbage collection, with the choice of object detection architecture being critical. This research employs the YOLOv4 model for real-time detection of marine debris using the Trash-ICRA 19 dataset, consisting of 7683 images at 480x320 pixels. Various modifications-pretrained models, training from scratch, mosaic augmentation, layer freezing, YOLOv4-tiny, and channel pruning-are compared to enhance architecture efficiency. Channel pruning significantly improves detection speed, increasing the base YOLOv4 frame rate from 15.19 FPS to 19.4 FPS, with only a 1.2% drop in mean Average Precision, from 97.6% to 96.4%.