首页|基于YOLOv5的试管检测算法

基于YOLOv5的试管检测算法

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

任克勤、陶青川

展开 >

四川大学电子信息学院,成都 610065

YOLOv5 ShuffleNetV2 轻量化 注意力机制

2022

现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
年,卷(期):2022.28(7)
  • 1
  • 9