公开(公告)号：CN118918418B

公开(公告)日：2024-12-31

申请号：CN202411407566.6

申请日：2024-10-09

Applicant: 杭州海康威视数字技术股份有限公司

Inventor： 郑秀枝 ,  张经纬 ,  许昀璐

IPC: G06V10/774 ,  G06V10/40 ,  G06V10/25 ,  G06V10/80

Abstract: 本申请实施例提供了一种目标检测模型的训练方法、装置及电子设备，涉及计算机技术领域，包括：获取多个样本数据集；将各样本数据集输入至待训练的目标检测模型中，利用目标检测模型的三维骨干网络提取点云数据的点云特征，利用目标检测模型的图像骨干网络提取图像数据的图像特征；针对每一样本数据集，对点云特征和图像特征进行特征融合，得到样本数据集的待检测特征；对各待检测特征进行目标检测，得到各样本数据集各自的目标检测结果；根据各样本数据集各自的感兴趣区域，对比各目标检测结果与各样本数据集的真值，确定目标检测模型的模型损失；基于模型损失对目标检测模型进行调整。本申请实施例提高了目标检测模型在不同数据集的泛化能力。

公开(公告)号：CN118918418A

公开(公告)日：2024-11-08

申请号：CN202411407566.6

申请日：2024-10-09

Applicant: 杭州海康威视数字技术股份有限公司

Inventor： 郑秀枝 ,  张经纬 ,  许昀璐

IPC: G06V10/774 ,  G06V10/40 ,  G06V10/25 ,  G06V10/80

Abstract: 本申请实施例提供了一种目标检测模型的训练方法、装置及电子设备，涉及计算机技术领域，包括：获取多个样本数据集；将各样本数据集输入至待训练的目标检测模型中，利用目标检测模型的三维骨干网络提取点云数据的点云特征，利用目标检测模型的图像骨干网络提取图像数据的图像特征；针对每一样本数据集，对点云特征和图像特征进行特征融合，得到样本数据集的待检测特征；对各待检测特征进行目标检测，得到各样本数据集各自的目标检测结果；根据各样本数据集各自的感兴趣区域，对比各目标检测结果与各样本数据集的真值，确定目标检测模型的模型损失；基于模型损失对目标检测模型进行调整。本申请实施例提高了目标检测模型在不同数据集的泛化能力。