公开(公告)号：CN118247638A

公开(公告)日：2024-06-25

申请号：CN202410204655.4

申请日：2024-02-24

Applicant: 西安电子科技大学

Inventor： 朱虎明 ,  缪孔苗 ,  王金成 ,  张凌云 ,  董西淼 ,  刘卉杰 ,  程新跃 ,  侯彪

IPC: G06V10/94 ,  G06V20/10 ,  G06V10/25 ,  G06V10/774 ,  G06V10/82 ,  G06N3/045 ,  G06N3/084

Abstract: 本发明公开一种遥感图像视觉定位模型的大批量数据并行训练方法，主要解决现有技术在百卡并行训练遥感图像视觉定位模型时检测精度低的问题。其实现步骤为：获取训练样本集；设计一个包括语言编码器、视觉编码器、多尺度特征融合网络和检测网络的遥感图像视觉定位模型；基于矩与学习率联动单周期调整对遥感图像视觉定位模型进行数据并行训练：1)计算学习率的更新值和一阶矩的更新值；2)计算当前时刻的学习率和一阶矩，平均各个设备的梯度并计算一阶动量和二阶动量；3)修正一阶动量和二阶动量；4)重复上述步骤直到模型收敛，得到训练好的遥感图像视觉定位模型。本发明减小了训练中期发散的情况，提高了模型的检测精度，可用于神经网络的深度学习。