基于单阶段目标检测算法的羊肉多分体实时分类检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2018YFD0700804)


Mutton Multipartite Real-time Classification and Detection Based on Single-stage Object Detection Algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对输送带场景中羊肉分体需要进一步分类检测问题,提出一种基于单阶段目标检测算法的羊肉多分体实时分类检测方法。在羊屠宰车间环境下采集包含多类、多个羊肉分体图像,经图像增广及归一化后建立羊肉多分体图像数据集,其中训练集7200幅,测试集1400幅,验证集400幅。利用单阶段目标检测算法YOLO v3引入迁移学习对羊肉多分体图像数据集展开训练并获得最优模型,基于最优模型返回图像中各羊肉分体的类别及其位置,从而实现羊肉分体的分类检测。选用平均精度及单幅图像平均处理时间作为评判模型检测精度与速度指标。然后通过更换羊肉多分体识别模型的特征提取网络优化检测速度。另外设置包含亮、暗两种亮度水平的附加光照数据集以及代表羊肉分体遮挡情形的附加遮挡数据集,分别验证优化后模型的泛化能力与抗干扰能力,并通过多尺度特征明显的颈部与腹肋肉测试优化后模型的鲁棒性。最后引入Mask R-CNN、Faster R-CNN、Cascade R-CNN和SSD 4种常用目标检测算法针对不同数据集分别进行对比试验,在此基础上,进一步更换特征提取网络为MobileNet V1、ResNet34和ResNet50验证优化后模型的综合检测能力。试验结果表明,优化后模型的检测速度较原始模型提升48.53%,同时对光照、遮挡复杂环境下羊肉多分体识别具备较强的泛化能力与抗干扰能力,以及对多尺度特征显著的羊肉分体检测具有良好的鲁棒性,针对羊肉多分体图像验证集,优化后羊肉多分体识别模型的平均精度达到88.05%,单幅图像处理时间为64.7ms,综合检测能力优于其他算法,说明该方法具备较高的检测精度和良好的实时性,能够满足实际生产需求。

    Abstract:

    Aiming at the problem that mutton multipartite needs to be further classified and detected in the conveyor belt scene, a real-time classification and detection method for mutton splits based on a single-stage object detection algorithm was proposed. In the sheep slaughter workshop environment, multiple types and multiple mutton split images were collected. After image augmentation and normalization, a mutton multipartite image data set was established, including 7200 training sets, 1400 test sets, and 400 verification sets. Using the single-stage object detection algorithm YOLO v3 to introduce transfer learning to train the mutton multipartite image data set and obtain the optimal model. Based on the optimal model, the category and position of each mutton split in the image were returned, so as to realize the classification and detection of mutton multipartite. The average accuracy mAP and the average detection time of a single image were selected as the accuracy and speed indicators for judging the detection effect of the model. Then, the detection speed was optimized by replacing the feature extraction network of the mutton multipartite recognition model. In addition, an additional illumination data set containing two brightness levels of “bright” and “dark” and an additional occlusion data set representing the occlusion situation of mutton were set to verify the generalization ability and anti-interference ability of the optimized model,and the robustness of the optimized model was tested through the neck and abdominal rib with obvious multi-scale features. Finally, four commonly used object detection algorithms: Mask R-CNN, Faster R-CNN, Cascade R-CNN, and SSD were introduced to conduct comparative experiments on different data sets. On this basis, the feature extraction network was further replaced with MobileNet V1, ResNet34 and ResNet50 to verify the optimized model’s comprehensive testing capabilities. The test results showed that the detection speed of the optimized model was 48.53% higher than that of the original model. At the same time, it had strong generalization ability and anti-interference ability for multi-part recognition of mutton under complex environment of light and shading, and it had good robustness to the mutton multi-part detection with multi-scale features. It was optimized for the verification set of mutton multipartite image. The mAP value of the optimization model reached 88.05%, and the processing time of a single image was 64.7ms, the comprehensive detection ability was better than that of other algorithms, which indicating that this method had high detection accuracy and good real-time performance, and can meet actual production needs.

    参考文献
    相似文献
    引证文献
引用本文

赵世达,王树才,郝广钊,张一驰,杨华建.基于单阶段目标检测算法的羊肉多分体实时分类检测[J].农业机械学报,2022,53(3):400-411. ZHAO Shida, WANG Shucai, HAO Guangzhao, ZHANG Yichi, YANG Huajian. Mutton Multipartite Real-time Classification and Detection Based on Single-stage Object Detection Algorithm[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(3):400-411.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-03-25
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-03-10
  • 出版日期: