[发明专利]一种模型训练的方法、图像检测的方法及装置

权利要求书

1.一种模型训练的方法，其特征在于，包括：

获取样本图像；

将所述样本图像输入到待训练的预测模型，确定所述样本图像中涉及指定目标物的至少一个候选图像区域，并确定每个候选图像区域对应的原始特征；

从所述至少一个候选图像区域中，筛选出包含其他目标物的候选图像区域，作为待处理区域；

根据所述其他目标物的图像在所述待处理区域中的位置，对所述待处理区域对应的原始特征进行补偿，得到所述待处理区域对应的补偿特征；

根据所述待处理区域对应的补偿特征以及其他候选图像区域对应的原始特征，得到针对所述样本图像的目标物检测结果，并以最小化所述目标物检测结果与所述样本图像中针对所述指定目标物的标签信息之间的偏差为优化目标，对所述预测模型进行训练。

2.如权利要求1所述的方法，其特征在于，所述预测模型包括：特征提取层、区域感知层；

将所述样本图像输入到待训练的预测模型，确定所述样本图像中涉及指定目标物的至少一个候选图像区域，并确定出每个候选图像区域对应的原始特征，具体包括：

将所述样本图像输入到所述特征提取层，确定所述样本图像对应的各尺寸的特征图；

针对每个尺寸的特征图，将该尺寸的特征图输入到所述区域感知层，确定该尺寸的特征图中涉及所述指定目标物对应的至少一个候选图像区域，并确定出在该尺寸的特征图下的每个候选图像区域对应的原始特征。

3.如权利要求2所述的方法，其特征在于，针对每个尺寸的特征图，将该尺寸的特征图输入到所述区域感知层，确定该尺寸的特征图中涉及所述指定目标物对应的至少一个候选图像区域，具体包括：

针对每个尺寸的特征图，将该尺寸的特征图输入到所述区域感知层，预测该尺寸的特征图中每个像素点属于所述指定目标物的第一类别概率；

将第一类别概率不小于设定概率阈值的像素点，作为目标像素点；

根据所述目标像素点，确定该尺寸的特征图中涉及所述指定目标物对应的至少一个候选图像区域。

4.如权利要求3所述的方法，其特征在于，以最小化所述目标物检测结果与所述样本图像中针对所述指定目标物的标签信息之间的偏差为优化目标，对所述预测模型进行训练，具体包括：

以最小化所述至少一个候选图像区域与所述区域感知层使用的第一标签信息之间的偏差，以及所述目标物检测结果与所述样本图像中针对所述指定目标物的标签信息之间的偏差为优化目标，对所述预测模型进行训练，所述第一标签信息用于表示候选图像区域是否为所述指定目标物在所述样本图像中的实际图像区域。

5.如权利要求1～4任一项所述的方法，其特征在于，所述预测模型包括：目标物分类层；

从所述至少一个候选图像区域中，筛选出包含其他目标物的候选图像区域，作为待处理区域，具体包括：

针对每个候选图像区域，将该候选图像区域进行图像放大，得到放大后的候选图像区域，所述放大后的候选图像区域包含有若干个图像网格；

将该候选图像区域对应的原始特征输入到所述目标物分类层，预测所述放大后的候选图像区域中，每个图像网格属于所述指定目标物的第二类别概率；

若确定所述放大后的候选图像区域中包含有第二类别概率小于设定概率阈值的图像网格，将该候选图像区域作为所述待处理区域。

6.如权利要求5所述的方法，其特征在于，以最小化所述目标物检测结果与所述样本图像中针对所述指定目标物的标签信息之间的偏差为优化目标，对所述预测模型进行训练，具体包括：

以最小化所述待处理区域与所述目标物分类层使用的第二标签信息之间的偏差，以及所述目标物检测结果与所述样本图像中针对所述指定目标物的标签信息之间的偏差为优化目标，对所述预测模型进行训练，所述第二标签信息用于标识出所述待处理区域中各目标物所处的实际图像区域。