现代计算机2022，Vol.28Issue(7) ：1-7,14.DOI:10.3969/j.issn.1007-1423.2022.07.001

基于YOLOv5的试管检测算法

Tube Detection Algorithm Based on YOLOv5

任克勤 陶青川
现代计算机2022，Vol.28Issue(7) ：1-7,14.DOI:10.3969/j.issn.1007-1423.2022.07.001
✕
  • NETL
  • NSTL
  • 维普
  • 万方数据

基于YOLOv5的试管检测算法

Tube Detection Algorithm Based on YOLOv5

任克勤 1陶青川1
扫码查看

作者信息

  • 1. 四川大学电子信息学院,成都 610065
  • 折叠

摘要

针对嵌入式平台上试管批量检测自动化任务提出了一种轻量化改进YOLOv5的试管检测算法,通过ShuffleNetV2构建YOLOv5的主干网,并且减少YOLOv5检测头的通道数量,大幅减少计算参数,通过将通道注意力与空间注意力机制混合,提升网络性能.实验在自建数据集上进行训练和测试,结果表明,改进的轻量化模型大幅降低了参数和计算量,同时保证了一定的精度,利于在嵌入式设备上进行部署.

关键词

YOLOv5/ShuffleNetV2/轻量化/注意力机制

引用本文复制引用

出版年

2022
现代计算机
中大控股

现代计算机

影响因子：0.292
ISSN：1007-1423
被引量1
参考文献量9
段落导航相关论文