目标检测方法和装置、电子设备以及存储介质
摘要文本
本发明实施例提供一种目标检测方法和装置、电子设备以及存储介质。该方法包括：获取待处理图像和经训练的目标检测模型，目标检测模型包括多个预测头，多个预测头与多个尺度范围一一对应，每个预测头用于对尺度处于对应尺度范围内的目标对象进行位置预测；基于待处理图像所对应的尺度信息，确定多个预测头中的待剪枝预测头，待剪枝预测头为多个预测头中除与一个或多个目标尺度范围对应的目标预测头外的剩余预测头；对目标检测模型中的待剪枝预测头进行剪枝，以获得剪枝后的目标检测模型；将待处理图像输入剪枝后的目标检测模型进行目标检测，以获得对应的目标检测结果。该方案可以解决预测头冗余的问题，提升目标检测模型的推理速度。
申请人信息
- 申请人:苏州镁伽科技有限公司
- 申请人地址:215000 江苏省苏州市中国(江苏)自由贸易试验区苏州片区苏州工业园区东平街277号
- 发明人: 苏州镁伽科技有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 目标检测方法和装置、电子设备以及存储介质 |
| 专利类型 | 发明授权 |
| 申请号 | CN202311776428.0 |
| 申请日 | 2023/12/22 |
| 公告号 | CN117456170B |
| 公开日 | 2024/3/19 |
| IPC主分类号 | G06V10/25 |
| 权利人 | 苏州镁伽科技有限公司 |
| 发明人 | 张共济; 曹高宇 |
| 地址 | 江苏省苏州市工业园区东平街277号 |
专利主权项内容
1.一种目标检测方法，其特征在于，所述方法包括：获取待处理图像和经训练的目标检测模型，所述目标检测模型包括多个预测头，所述多个预测头与多个尺度范围一一对应，每个预测头用于对尺度处于对应尺度范围内的目标对象进行位置预测；基于所述待处理图像所对应的尺度信息，确定所述多个预测头中的待剪枝预测头，其中，所述尺度信息用于指示预期从所述待处理图像中检测的所述目标对象的一个或多个目标尺度范围，所述待剪枝预测头为所述多个预测头中除与所述一个或多个目标尺度范围对应的目标预测头外的剩余预测头；对所述目标检测模型中的所述待剪枝预测头进行剪枝，以获得剪枝后的目标检测模型；将所述待处理图像输入所述剪枝后的目标检测模型进行目标检测，以获得对应的目标检测结果；其中，所述基于所述待处理图像所对应的尺度信息，确定所述多个预测头中的待剪枝预测头，包括：获取目标数据集，所述目标数据集包括多个目标图像，所述多个目标图像中所述目标对象的尺度是已知的，所述待处理图像中所述目标对象的尺度范围默认为与所述多个目标图像中的所述目标对象的尺度范围一致；基于所述目标数据集中的所述目标对象在所述多个尺度范围中的分布情况，确定所述待剪枝预测头，其中，所述尺度信息包括所述目标数据集中的所述目标对象在所述多个尺度范围中的分布情况；其中，所述基于所述目标数据集中的所述目标对象在所述多个尺度范围中的分布情况，确定所述待剪枝预测头，包括：对于所述多个尺度范围中的每个尺度范围，计算所述目标数据集中所述目标对象的尺度处于该尺度范围内的目标图像的数量与所述多个目标图像的总数量之间的比值，所述分布情况采用所述比值表示；在所述比值小于或等于预设比例阈值时，将该尺度范围所对应的预测头确定为所述待剪枝预测头。 马-克-数据