Beach waste detection is a critical step toward addressing environmental pollution and preserving marine ecosystems. In this research, we explore the application of object detection models for the identification of beach waste in complex environments. We took two popular architectures, YOLOv8n and YOLOv11n, and fine-tuned them on a custom dataset. Later, we proposed a transfer learning-based modified YOLOv11n model by replacing the default SiLU activation function with the ReLU activation function in the convolution layers. Training the proposed model required significantly less time-3.207 hours compared to 3.753 hours for the original YOLOv11n model-representing a 14.55% improvement in training speed. The proposed model attained a mean average precision at 50% IoU (mAP50) of 0.831 and a precision of 0.846 on the test set. These findings emphasize the potential of ReLU activation function in YOLOv11n for faster training in object detection tasks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Time-Efficient Transfer Learning Approach for Waste Detection

  • Chinmay Bhangale,
  • Prajwal Gabhane,
  • M. Anand Kumar

摘要

Beach waste detection is a critical step toward addressing environmental pollution and preserving marine ecosystems. In this research, we explore the application of object detection models for the identification of beach waste in complex environments. We took two popular architectures, YOLOv8n and YOLOv11n, and fine-tuned them on a custom dataset. Later, we proposed a transfer learning-based modified YOLOv11n model by replacing the default SiLU activation function with the ReLU activation function in the convolution layers. Training the proposed model required significantly less time-3.207 hours compared to 3.753 hours for the original YOLOv11n model-representing a 14.55% improvement in training speed. The proposed model attained a mean average precision at 50% IoU (mAP50) of 0.831 and a precision of 0.846 on the test set. These findings emphasize the potential of ReLU activation function in YOLOv11n for faster training in object detection tasks.