Performance Evaluation of YOLO-Based Deep Learning Models for Real-Time Armour Unit Detection with Image Pre-processing Method
Firmansyah Putra Pratama, Alfan Rizaldy Pratama, Dewi Mutiara Sari, Bayu Sandi Marta, R. Haryo Dwito Armono · 2024
Breakwater construction in Indonesia still relies on divers to direct the placement of rock armour units, which is risky and time-constrained. This research aims to replace the diver's task with a deep learning-based vision system using YOLO-based deep learning models. The system utilizes image pre-processing technology by applying histogram equalization (HE) techniques to improve image quality before the detection process. This research evaluates the performance of the YOLO-based deep learning models in detecting armour units in real-time with a focus on various environmental conditions, which are clear and murky water. The analysis reveals clear water consistently supports higher average frame rates (FPS) compared to murky water, maintaining efficient frame processing across all models. In murky water, histogram equalization significantly enhances detection accuracy from 60% to 80% for YOLOv4-tiny and YOLOv7-tiny, demonstrating its effectiveness in challenging conditions. Notably, accuracy remains at 100% for all models in clear water, underscoring their robust performance under optimal visibility conditions.