摘要
本发明公开了一种改进的轻量化YOLOv4人脸口罩检测方法,针对实际有效感受野远小于理论感受野,当目标占据整张图片一半以上大小时,由于实际有效感受野的限制,网络的检测效果不好;新设计的SPP模块由四个池化层组成,将输入特征图分别经过四层池化,每一层池化后的输出特征图的感受野进一步扩大,同时也将池化后的多张不同感受野的特征图进行特征融合操作,实现了局部高级语义特征的多尺度利用;将新设计的SPP模块嵌入至YOLOv4-Tiny骨架网络后面,FPN结构前面,构成SPP-YOLOv4-light网络,利用SPP模块将网络深层的感受野进一步扩大,同时将骨架网络提取到的特征进行多尺度池化后融合,在通用数据集上进行训练和测试,验证了本发明方法提高了人脸口罩的检测精度。
- 单位