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

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

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

     

    Abstract: A mass spectrometer experimental instrument robotic arm automatic recognition and detection method based on improved YOLOv5s is proposed to address the issues of improving recognition accuracy and speed, as well as reducing system costs in the recognition technology of grabbing samples for mass spectrometer robotic arms. Firstly, the CA attention mechanism is introduced, which can strengthen the feature learning and feature extraction of the sample tray image, while reducing the impact of the sample surface background on the detection results, especially the background similar to the sample tray. Secondly, a path is added to the original backbone network structure that directly connects the shallow feature map to the deep layer to enhance the model’s detection ability of small target objects; finally, the original C3 module is replaced with the DSConv module to reduce the complexity of the model, and maintain its accuracy. The experimental results show that in the custom dataset, compared to YOLOv5s, the improved model has slightly increased parameter and computational complexity, significantly improved detection accuracy. In the COCO public dataset, the model proposed in this paper outperforms other models with high accuracy of 0.723 mAP, while maintaining a compact weight size of 20.2 MB and a real-time processing speed of 35.81 FPS.

     

/

返回文章
返回