计算机与现代化

• 图像处理 • 上一篇    

基于Faster R-CNN模型的传送带药盒与空位检测方法

  

  1. (1.MIT Laboratory for Financial Engineering, Cambridge, MA 02139, USA;2.南开大学计算机学院,天津300350)
  • 收稿日期:2019-07-21 出版日期:2019-09-23 发布日期:2019-09-23
  • 作者简介:张瑞勋(1988-),男,浙江杭州人,数据科学家,研究方向:数据科学,金融工程,E-mail: zhangruixun@gmail.com; 邵秀丽(1963-),女,江苏南通人,教授,博士生导师,研究方向:人工智能,数据分析,软件工程,E-mail: shaoxl@nankai.edu.cn。
  • 基金资助:
    天津市智能制造专项资金资助项目(201707105,201907210); 天津市互联网先进制造专项资金资助项目(18ZXRHGX00110)

A Detection Method of Medicine Box and Vacancy on Conveyor Based on Faster R-CNN Model

  1. (1. MIT Laboratory for Financial Engineering, Cambridge, MA 02139, USA;
    2. College of Computer Science, Nankai University, Tianjin 300350, China)
  • Received:2019-07-21 Online:2019-09-23 Published:2019-09-23

摘要: 制药企业为了判断传送带药盒的拥堵情况,需要对传送带上的药盒和空位进行定位,但人工方式效率低下,实时性差。在此背景下,结合Faster R-CNN模型,提出传送带目标检测方法。基于传送带图像构建模型训练集和测试集,将训练集通过ZFNet卷积神经网络计算卷积特征,并利用RPN(Region Proposal Network)生成精准的候选区域,在此基础上基于Faster R-CNN模型在候选区域上进行分类和回归,计算得到药盒与空位矩形框。通过使用测试集测试模型进行目标标注并计算出概率,结果表明,本方法对传送带目标的检测准确率良好。

关键词: Faster R-CNN, ZFNet卷积神经网络, 目标检测

Abstract: In order to find the congestion of the conveyor, the pharmaceutical company needs to locate the medicine boxes and vacancies on the conveyor, but the manual method is of inefficiency and has poor real-time performance. In this context, combining with the Faster R-CNN model, a target detection method is proposed. In this method, training set and testing set are constructed based on the conveyor images, then, the training set is processed through the ZFNet convolutional neural network to calculate the convolution characteristics, and the RPN (Region Proposal Network) is used to generate accurate candidate regions. On this basis, the classification and regression based on the Faster R-CNN model are performed on the candidate regions, and the rectangular boxes of the kit and the vacancy are calculated. At last, the trained model is tested by using the testing set to label the target and to calculate the probability. The result shows that the method has good detection accuracy for the conveyor belt target.

Key words: Faster R-CNN, ZFNet convolutional neural network, target detection

中图分类号: