Tencent / FaceDetection-DSFD

腾讯优图高精度双分支人脸检测器
Other
2.91k stars 728 forks source link

数据增强的疑问 #38

Open oftenliu opened 5 years ago

oftenliu commented 5 years ago
  1. 代码段augmentations.py:412~416行
    centers = (boxes[:, :2] + boxes[:, 2:]) / 2.0 m1 = (choice_box[0] < centers[:, 0]) (choice_box[1] < centers[:, 1]) m2 = (choice_box[2] > centers[:, 0]) (choice_box[3] > centers[:, 1]) mask = m1 * m2#是否包含某个box的中心点 #这个保留face box的依据带考究 只要中心点 在sample_boxes中就保存 current_boxes = boxes[mask, :].copy() 通过判断源face box的中心点是否在crop区域来确定该box是否保留?这个原则是否会出现一些box中人脸占比很少的情况?