一种轻量化煤矸目标检测方法

摘要

本发明公开了一种轻量化煤矸目标检测方法，具体为：首先，在不同光照条件下采集煤矸图片，建立煤矸数据集；利用轻量化特征提取网络，提取煤矸图片的多尺度特征；再应用自注意力分支对浅层特征图进行特征增强；之后利用修改后的锚框参数生成每种尺度特征图对应的锚框；以SSD目标检测模型为基本框架，构建改进SSD煤矸目标检测模型；最后，采用煤矸训练集对模型进行训练。本发明的方法，实现了检测精度和速度更好的权衡，通过对模型进行轻量化处理、引入自注意力机制以及锚框优化，提高了煤矸目标检测的精度和速度。