[发明专利]物体检测方法、模型训练方法、装置及电子设备

权利要求书

1.一种物体检测方法，包括：

获取待检测图像；

对所述待检测图像进行物体检测，得到所述待检测图像的第一物体检测信息；

对所述待检测图像进行场景检测，以获取所述待检测图像的目标场景检测信息，所述目标场景检测信息包括所述待检测图像对应的目标场景类别；

获取所述目标场景类别对应的场景识别模型，所述场景识别模型用于获取所述目标场景类别下的场景识别信息，所述场景识别信息用于表征所述目标场景类别对应的场景框中物体的场景环境；

基于所述场景识别模型和所述第一物体检测信息，确定所述待检测图像的第二物体检测信息，所述第二物体检测信息用于表征物体在所述目标场景类别的场景下的摆放信息；

所述基于所述场景识别模型和所述第一物体检测信息，确定所述待检测图像的第二物体检测信息，包括：

基于所述场景识别模型对所述待检测图像进行场景识别，得到所述目标场景类别下的场景识别信息；

对所述第一物体检测信息和所述场景识别信息进行融合处理，得到所述第二物体检测信息；

所述基于所述场景识别模型对所述待检测图像进行场景识别，得到所述目标场景类别下的场景识别信息，包括：

将标记有所述目标场景类别对应的场景框的待检测图像输入至场景识别模型进行场景识别，得到所述目标场景类别下的场景识别信息；

所述对所述待检测图像进行场景检测，以获取所述待检测图像的目标场景检测信息，包括：

基于场景检测模型对所述待检测图像进行场景检测，以获取所述待检测图像的目标场景检测信息；所述场景检测模型为基于行业场景库中行业类别下的场景图像样本数据训练得到的模型，所述行业场景库包括M个行业类别下的场景图像样本数据，M为正整数；

所述基于场景检测模型对所述待检测图像进行场景检测，以获取所述待检测图像的目标场景检测信息之前，所述方法还包括：

获取所述待检测图像对应的目标行业类别；

从预设场景检测模型中获取所述目标行业类别对应的所述场景检测模型。

2.根据权利要求1所述的方法，其中，所述行业场景库包括每个行业类别对应的N个场景类别下的场景图像样本数据，N为正整数，所述场景识别模型为基于目标数据训练得到的模型，所述目标数据包括所述行业场景库中所述目标行业类别的所述目标场景类别下的场景图像样本数据，和/或，输入的所述目标场景类别下的物体图像样本数据。

3.一种模型训练方法，包括：

获取目标数据，所述目标数据包括目标场景类别下的场景图像样本数据，所述目标场景类别为对待检测图像进行场景检测所获取的目标场景检测信息中的场景类别；

基于所述目标数据对所述目标场景类别对应的场景识别模型进行训练；

其中，所述场景识别模型用于结合第一物体检测信息确定所述待检测图像的第二物体检测信息，所述第一物体检测信息为对所述待检测图像进行物体检测得到的物体检测信息，所述场景识别模型用于获取所述目标场景类别下的场景识别信息，所述场景识别信息用于表征所述目标场景类别对应的场景框中物体的场景环境，所述第二物体检测信息用于表征物体在所述目标场景类别的场景下的摆放信息；

所述待检测图像的第二物体检测信息通过以下方式确定：

基于所述场景识别模型对所述待检测图像进行场景识别，得到所述目标场景类别下的场景识别信息；

对所述第一物体检测信息和所述场景识别信息进行融合处理，得到所述第二物体检测信息；

所述基于所述场景识别模型对所述待检测图像进行场景识别，得到所述目标场景类别下的场景识别信息，包括：

将标记有所述目标场景类别对应的场景框的待检测图像输入至场景识别模型进行场景识别，得到所述目标场景类别下的场景识别信息；

所述目标场景类别为基于场景检测模型对待检测图像进行场景检测所获取的目标场景检测信息中的场景类别，所述目标场景类别为目标行业类别下的场景类别，所述目标行业类别为所述待检测图像对应的行业类别；

所述方法还包括：

从行业场景库中获取所述目标行业类别下的场景图像样本数据；

基于所述目标行业类别下的场景图像样本数据对所述场景检测模型进行训练，其中，所述行业场景库包括M个行业类别下的场景图像样本数据，所述M个行业类别包括所述目标行业类别，M为正整数。