人体检测框和人脸框匹配问题

原创

已于 2023-07-01 16:58:00 修改 · 753 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #匈牙利算法 #人脸框和人体框匹配

于 2023-07-01 16:55:32 首次发布

该文介绍了在人脸识别和表情识别系统中，如何通过计算人体检测框和人脸检测框的IoU来确定匹配关系，并使用匈牙利算法优化这一过程，以最小化匹配损失。

背景

需要知道照片中人的位置及相应的人脸的位置。使用人体检测器和人脸检测器。
使用人脸检测器，方便做后续的人脸识别和表情识别等

步骤：

计算所有人体检测框和人脸框的iou
计算iou_loss。iou_loss = 1 - iou
通过匈牙利算法得到人脸框和人体框之间的匹配关系。匈牙利算法使得分配问题的和最小

代码实现

### 计算人脸框和人体框间的iou
def iou_match_grid(bbox1, bbox2):
    ### bbox1,bbox2的表示形式为[xmin,ymin,xmax,ymax]
    bbox1 = np.array(bbox1)
    bbox2 = np.array(bbox2)
    bbox1 = bbox1[:, :4]
    bbox2 = bbox2[:, :4]

    M = bbox1.shape[0]
    N = bbox2.shape[0]

    bbox1 = np.expand_dims(bbox1, axis=1) ## [M,1,4]
    bbox1 = np.repeat