模型训练方法、目标检测方法、电子设备及存储介质

    公开(公告)号:CN116597246A

    公开(公告)日:2023-08-15

    申请号:CN202310403529.7

    申请日:2023-04-14

    Abstract: 本发明提供一种模型训练方法、目标检测方法、电子设备及存储介质,其中方法包括:获取原始数据集,所述原始数据集中每个原始数据包括对应视角下拍摄的原始图像和所述原始图像中感兴趣目标的种类信息和位置信息;对所述原始数据集进行数据筛选和数据补充,确定第一目标数据集;基于所述第一目标数据集,对初始透视变换矫正网络模型进行训练,确定目标透视变换矫正网络模型;基于所述目标透视变换矫正网络模型,对所述原始数据集进行数据补全,确定第二目标数据集;基于所述第二目标数据集,对初始目标检测网络模型进行训练,确定预设目标检测网络模型。本发明在无需获取新数据扩充原始数据集时也能够准确识别大视角或者小视角图像中的感兴趣目标。

Patent Agency Ranking