一种轻量化可见光绝缘子目标检测方法及系统
摘要:
本发明公开了一种轻量化可见光绝缘子目标检测方法及系统,方法包括:构建SSD检测网络模型;将待检测的可见光图像输入至目标检测网络模型中,目标检测网络模型通过主干特征提取网络和附加层对可见光图像进行特征提取,提取出不同尺度的特征图,并通过至少一个检测头预测不同尺度的特征图中每个位置上存在的目标的类别和位置,得到与至少一个检测头对应的至少一个输出结果;将至少一个输出结果映射至待检测的可见光图像上进行非极大值抑制,得到目标输出结果。采用无锚框检测头,不需要预定义和生成锚框,通过直接回归目标的中心点和边界框,可以更准确地定位目标同时更好地适应不同尺度和形状的目标。
公开/授权文献
0/0