提出一种基于改进YOLOv5的轻量型口罩检测算法CG-YOLOv5s。结合卷积注意力机制和Ghost卷积等技术，在网络中加入CBAM-CSP、GRCM改进模块以增强特征提取能力，改善模型的计算性能，使用Alpha-CIoU损失并结合DIoU非极大抑制方法，进一步提升检测精度。结果表明，CG-YOLOv5s在对检测速度影响较小的情况下，获得了89.1%的检测精度，模型大小减少了19.63%，实现了模型轻量化的效果。
科研之友
微信
新浪微博
Facebook
分享链接
