原创力.专利
发明
目标检测模型训练方法、目标检测方法、装置及电子设备2025
2025-05-24 13:05:29
发布于四川
1
- 申请专利号：CN202510098188.6
- 公开（公告）日：2025-05-23
- 公开（公告）号：CN120032332A
- 申请人：浙江吉利控股集团有限公司|||重庆睿蓝汽车研究院有限公司|||重庆睿蓝汽车科技有限公司
摘要：本发明提供了一种目标检测模型训练方法、目标检测方法、装置及电子设备，涉及目标识别脱敏技术领域。本发明所述的目标检测模型训练方法，包括：对视频数据集进行预处理，以生成图像数据集；将所述图像数据集输入到预设YOLOv5模型中进行训练，以建立目标检测模型，其中，所述预设YOLOv5模型采用改进损失函数，所述改进损失函数包括IOU损失、中心点损失和长宽损失。本发明可以使预设YOLOv5模型预测的目标盒与图像数据集中的锚盒的宽高度的差值最小化，从而使收敛速度更快，定位效果更优，且具有更稳定的训练过程和更优的定位精度，进而能够提高目标检测模型训练效率和识别精度。
专利内容
本发明提供了一种目标检测模型训练方法、目标检测方法、装置及电子设备，涉及目标识别脱敏技术领域。本发明所述的目标检测模型训练方法，包括：对视频数据集进行预处理，以生成图像数据集；将所述图像数据集输入到预设YOLOv5模型中进行训练，以建立目标检测模型，其中，所述预设YOLOv5模型采用改进损失函数，所述改进损失函数包括IOU损失、中心点损失和长宽损失。本发明可以使预设YOLOv5模型预测的目标盒与图像数据集中的锚盒的宽高度的差值最小化，从而使收敛速度更快，定位效果更优，且具有更稳定的训练过程和更优的定位精度，进而能够提高目标检测模型训练效率和识别精度。G06V20/56(2022.01);G06V10/774(2022.01);G06V10/82(2022.01);G06N3/08(2023.01)
