重庆理工大学学报(自然科学) ›› 2021, Vol. 35 ›› Issue (6): 174-184.doi: 10.3969/j.issn.1674-8425(z).2021.06.022

• 信息·计算机 • 上一篇    

基于 YOLOGT网络的零售商品目标检测方法

周可鑫,左云波,谷玉海,朱腾腾,卞景艺   

  1. 北京信息科技大学 现代测控技术教育部重点实验室,北京 100192
  • 收稿日期:2020-06-23 发布日期:2021-07-20
  • 作者简介:周可鑫,男,硕士研究生,主要从事图像处理与目标检测研究,Email:13261972081@163.com。
  • 基金资助:
    国家自然科学基金项目(51975058);北京市教委促进高校内涵发展项目(5112011015)

  • Received:2020-06-23 Published:2021-07-20

摘要: 提出改进的 YOLOGT深度学习网络,实现基于嵌入式平台的零售商品数据集 RPC 的图像目标检测。YOLOGT网络以 YOLOv3Tiny网络为基础,采用线性扩展优化用于特征提 取的卷积层,并将深度学习网络浅层信息与深层信息融合使得网络成为三尺度预测网络,提高 网络对小目标商品的识别能力。YOLOGT采用 Mish函数优化了网络激活函数,并重新利用 FCM聚类算法生成多个新的先验框。利用 RPC零售商品数据集在嵌入式平台 Jetsonnano核心 板上实验,YOLOGT网络对于每张图片的平均检测速度为 342.67ms/张,平均检测准确率 (mAP)为 97.21%,与 YOLOv3Tiny相比,检测速度提升了 57.81%,mAP提升了 1.84%,并降 低了小目标漏检率,能够满足使用要求。

关键词: 深度学习, 目标检测, RPC, 嵌入式, YOLOGT

中图分类号: 

  • TP391.41