基于改进YOLOv5s的质谱实验仪器机械臂样品盘检测优化

Optimization of Sample Plate Detection for Robotic Arm in Mass Spectrometry Experiment Instrument Based on Improved YOLOv5s

  • 摘要: 针对质谱实验仪器机械臂在抓取待检测样品的识别技术中提高识别的准确率和速度以及降低系统成本问题,提出一种基于改进 YOLOv5s 的质谱实验仪器机械臂自动识别检测方法。首先,引入 CA 注意力机制,可以加强对样品盘图像的特征学习和特征提取,同时减弱样品表面背景对检测结果的影响,特别是与样品盘相似的背景;其次,在原主干网络结构中增加一条从浅层特征图直接连接到深层的路径,以增强模型对小目标物体的检测能力;最后,将原有的C3模块替换成DSConv模块,减小模型的复杂度并保持其准确性。实验结果表明,在自定义数据集中,最终相比于YOLOv5s,改进模型的参数量和运算量有所增加,检测精度显著提升。在Common Objects in Context(COCO)公共数据集中,改进的模型取的0.723的mAP精度,检测性能显著优于对比模型;模型权重大小仅为20.2 MB,实时处理速度为35.81 FPS,具备轻量化与实时检测优势。

     

    Abstract: To address the challenges of improving recognition accuracy and speed while reducing system costs in the recognition technology for robotic arms grasping samples to be tested in mass spectrometer experimental instruments, this paper proposes an automatic recognition and detection method based on improved YOLOv5s. Firstly, a Coordinate Attention (CA) mechanism is introduced to enhance feature learning and extraction from sample plate images, while reducing the impact of the background of the sample plate surface on detection results, especially when it is similar to the sample plate. Secondly, a path that directly connects shallow feature maps to deep feature maps is added to the original backbone network structure to enhance the model’s capability of detecting small target objects. Finally, the original C3 module is replaced with the DSConv module to reduce model complexity and maintain its accuracy. Experimental results show that in the custom dataset, compared to YOLOv5s, the improved model exhibits slightly increased parameter count and computational load but achieves significantly higher detection accuracy. In the COCO public dataset, the model outperforms other models with a high mAP of 0.723, while maintaining a compact model size of 20.2 MB and a real-time processing speed of 35.81 FPS.

     

/

返回文章
返回