一种基于改进YOLOv4的电网基建目标检测方法
摘要:
一种基于改进YOLOv4的电网基建目标检测方法,由主干特征提取网络CSPDarknet53得到的多尺度特征图先输入到ASPP模块采样,再输入到颈部网络PANet中,先经过FPN层进行上采样,FPN层的上采样结果一边回传至主干特征提取网络CSPDarknet53以多次观察图像以获得更丰富的特征信息,另一边则输入PANet网络进行下采样,PANet网络的下采样结果与FPN层的上采样结果进行特征双向融合,特征双向融合得到的特征图输入到头部预测网络Head中执行目标检测任务。本发明将FPN层每一层采样结果回传给主干特征提取网络CSPDarknet53以多次观察图像以获得更丰富的特征信息,提高针对某一特定目标的特征提取效率,最终提高检测效率。
公开/授权文献
0/0