有效

目标检测网络的训练方法、电子设备及存储介质

牛新、何贤浩、窦勇、姜晶菲、李荣春、苏华友、乔鹏、潘衡岳
中国人民解放军国防科技大学

摘要

本申请公开一种目标检测网络的训练方法、电子设备及存储介质,该方法包括:分别构建孪生网络和目标检测网络的网络结构;获取训练集,训练集中包括多组含有飞行器图像的交叉样本图像;根据训练集包括的交叉样本图像训练孪生网络,得到训练后的孪生网络对应的网络参数;根据网络参数和训练集包括的交叉样本图像,训练目标检测网络。本申请根据训练好的孪生网络的网络参数和训练集来训练目标检测网络。由于孪生网络的网络参数对于不同样本具有很好的鲁棒性,因此将训练好的孪生网络的网络参数作为目标检测网络的初始化参数,能够使得目标检测网络在初始状态具备较好的鲁棒性,从而训练后能够提高目标检测的准确性。