[发明专利]目标框预测方法、装置、设备及介质

权利要求书

1.一种目标框预测方法，其特征在于，包括：

对图像中预设类别的多个目标进行检测，得到多个候选预测框；

确定所述多个候选预测框分别与所述预设类别对应的各个标注框之间的重合度；

从所述各个候选预测框中，确定与所述多个标注框匹配且使得重合度之和最大的多个匹配预测框；

根据所述多个匹配预测框与所述多个标注框之间的匹配关系，确定所述预设类别的多个目标在所述图像中匹配的目标预测框。

2.根据权利要求1所述的方法，其特征在于，从所述各个候选预测框中，确定与所述多个标注框匹配且使得重合度之和最大的多个匹配预测框，包括：

基于所述多个候选预测框以及所述各个标注框，构建第一EMD模型，其中，以每个候选预测框为一个节点，每个标注框为一个节点，构建每个候选预测框所在的节点到每个标注框所在的节点之间的边，所述第一EMD模型包括的各个边的流表征所连接的两个节点所表征的两个框之间匹配的概率，所述第一EMD模型包括的各个边的代价表征所连接的两个节点所表征的两个框之间的重合度；

对所述第一EMD模型求解，确定与所述多个标注框匹配且使得重合度之和最大的多个匹配预测框。

3.根据权利要求2所述的方法，其特征在于，对所述第一EMD模型求解，确定与所述多个标注框匹配且使得重合度之和最大的多个匹配预测框，包括：

求解以下线性规划方程，以确定与所述多个标注框匹配且使得重合度之和最大的多个匹配预测框：

其中，f_ij≥0 i＝1，2，...，m，j＝1，2，...，n；

其中，所述线性规划方程的约束条件为：

其中：s_i＝p_i i＝1，2，...，m

d_j＝1 j＝1，2，...，n

c_ij＝1-GIoU(b1_i，b2_j)i＝1，2，...，m，j＝1，2，...，n

其中，m表示所述多个候选预测框的总数，n表示所述各个标注框的总数，f_ij和c_ij分别表示第i各个候选预测框所在的节点与第j个标注框所在的节点之间的边的流和代价，pi表示第i个候选预测框所框选的目标的类别预测概率中所述预设类别的预测概率，bi表示第i个候选预测框所框选的区域，b2_j表示第j个标注框所框选的区域。