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