• YOLOv8-REM:一种基于显微镜图像的水体微塑料实时检测算法

    YOLOv8-REM: a real-time microplastic detection algorithm for water based on microscopic images

    • 针对依赖扫描电子显微镜的深度学习检测方法因成像时间长导致效率低下、检测算法难以在实时性、检测效率与精度之间取得平衡,以及微塑料检测中缺乏适用于光学显微镜图像的科学深度学习数据集等问题,提出了基于YOLOv8s算法框架改进得到的YOLOv8-REM算法,并构建了一个科学的微塑料光学显微镜图像数据集。在算法的改进方法中,首先,通过减少YOLOv8s骨干网络层数,以减少深层卷积对微小目标信息的损失,防止小目标信息淹没在深层特征中,使网络更聚焦于浅层特征,从而提升了网络对小目标的检测能力与运行速度。此外,引入高效多尺度注意力模块。该模块利用指数加权的方法通过对特征图进行加权平均,强调重要特征并抑制不重要特征,逐步更新特征图的表示,使得模型能够更好地捕捉长期依赖关系和动态变化,提高模型对关键信息的关注度,从而增强模型对微小目标特征的捕捉能力。最后,设计多尺度特征融合模块,通过融合多尺度特征来补充小目标信息,进一步优化了检测性能。实验结果表明:YOLOv8-REM算法的平均检测精度(mAP@0.5:0.95)达到78.4%,较原始YOLOv8s提升5.2%,参数量减少27.03%。在保证实时性的前提下,该方法实现了对显微镜图像水体微塑料的高效、精准检测。

       

      Abstract: To address the issues of low efficiency caused by the long imaging time of scanning electron microscopes in deep learning-based detection methods, the difficulty in balancing real-time performance, detection efficiency, and accuracy in detection algorithms, and the lack of a scientific deep learning dataset suitable for optical microscope images in microplastic detection, the YOLOv8-REM algorithm is proposed, which is an improvement based on the YOLOv8s framework, and constructs a scientific optical microscope image dataset for microplastics. In the algorithm improvement, firstly, by reducing the number of layers in the YOLOv8s backbone network, the loss of small-object information due to deep convolutions is minimized, preventing small-object features from being overshadowed by deep-layer features. This allows the network to focus more on shallow features, thus enhancing both small-object detection capability and processing speed. Additionally, an efficient multi-scale attention module is introduced, which uses exponential weighting to perform weighted averaging of feature maps, emphasizing important features and suppressing irrelevant ones. This gradually updates the representation of the feature map, enabling the model to better capture long-term dependencies and dynamic changes, increasing its focus on key information and enhancing its ability to detect small-object features. Finally, a multi-scale feature fusion module is designed to supplement small-object information by integrating features from multiple scales, further optimizing detection performance. Experimental results show that the YOLOv8-REM algorithm achieves an average detection precision (mAP@0.5:0.95) of 78.4%, a 5.2 percentage point improvement over the original YOLOv8s, and a 27.03% reduction in parameter count. While ensuring real-time performance, this method achieves efficient and accurate detection of microplastics in water based on optical microscope images.

       

    /

    返回文章
    返回