[发明专利]一种目标检测方法、装置、电子设备及存储介质

说明书

本发明公开了一种目标检测方法，包括：对待检测目标进行三维建模，得到待检测目标的三维数字模型；利用三维数字模型生成目标图像，并利用目标图像生成数据集；利用数据集对目标检测网络进行训练，并利用训练后的目标检测网络对待检测目标进行目标检测；在本方法中，待检测目标的目标图像利用待检测目标的三维数字模型进行自动生成，无需人工实拍，可有效提升目标图像采集效率并降低采集成本；同时，该三维数字模型由待检测目标经过三维建模得到，可有效确保利用三维数字模型生成的目标图像能够用于目标检测网络的训练，并可最终提升目标检测网络的精准度；本发明还提供一种目标检测装置、电子设备及存储介质，具有上述有益效果。

技术领域

本发明涉及目标检测领域，特别涉及一种目标检测方法、装置、电子设备及存储介质。

背景技术

随着目标检测技术的不断发展，目标检测方法在侦查、预警等军事领域中的应用逐渐增多。为了实现对待检测目标的良好检测，需要事先为待检测目标准备大量的目标图像，并利用目标图像对目标检测网络进行训练，从而得到该待检测目标的目标检测网络。

相关技术中，待检测目标的目标图像依靠人工实拍进行采集，人工实拍受时间、人力及经济成本等因素的制约，难以快速获取大量目标图像，进而采用人工实拍的方式进行目标图像采集将严重影响目标检测网络的检测效果。

发明内容

本发明的目的是提供一种目标检测方法、装置、电子设备及存储介质，可利用待检测目标的三维数字模型生成待检测目标的目标图像，无需人工实拍，可有效提升目标图像采集效率并降低采集成本。

为解决上述技术问题，本发明提供一种目标检测方法，包括：

对待检测目标进行三维建模，得到所述待检测目标的三维数字模型；

利用所述三维数字模型生成目标图像，并利用所述目标图像生成数据集；

利用所述数据集对目标检测网络进行训练，并利用训练后的目标检测网络对所述待检测目标进行目标检测。

可选地，所述利用所述三维数字模型生成目标图像，并利用所述目标图像生成数据集，包括：

调整所述三维数字模型的模型参数，得到具有不同模型参数的三维数字模型；

利用所述具有不同模型参数的三维数字模型生成所述目标图像；

对所述目标图像进行标注，并依照预设数据格式利用标注后的目标图像生成数据集。

可选地，所述目标检测网络为YOLO目标检测网络。

可选地，所述对待检测目标进行三维建模，得到所述待检测目标的三维数字模型，包括：

对所述待检测目标进行倾斜摄影，得到所述待检测目标的模型图像；

将所述模型图像输入倾斜摄影建模软件进行计算，得到所述待检测目标的三维数字模型。

可选地，所述对所述待检测目标进行倾斜摄影，得到所述待检测目标的模型图像，包括：

对所述待检测目标的实物模型进行倾斜摄影，得到所述待检测目标的模型图像。

可选地，所述对所述待检测目标进行倾斜摄影，得到所述待检测目标的模型图像，包括：

将所述模型图像的图像中心设置为所述待检测目标，并将图像采集设备、所述待检测目标及水平面构成的拍摄角度设置为第一预设值；

将所述模型图像的图像中心设置为所述待检测目标，将图像采集设备的拍摄角度设置为第一预设值，并将相邻所述模型图像的航向重叠度设置为第二预设值，将相邻所述模型图像的旁向重叠度设置为第三预设值；

依照所述图像中心、所述拍摄角度、所述航向重叠度及所述旁向重叠度，对所述待检测目标进行环绕拍摄，得到初始模型图像；

将拍摄角度调整为第四预设值，并执行所述依照所述图像中心、所述拍摄角度、所述航向重叠度及所述旁向重叠度，对所述待检测目标进行环绕拍摄的步骤，直至所有所述初始模型图像完全覆盖所述待检测目标；