数据官网:http://shuoyang1213.me/WIDERFACE/
官方训练集和验证集都有标注,测试集没有
验证集:
WIDER_val.zip
标注:wider_face_val_bbx_gt.txt
标注文件:wider_face_split.zip 官网有。
retinaface网址:
https://github.com/biubug6/Pytorch_Retinaface
验证工具:
Windows也可以编译
这里面有验证结果标注。
http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/support/eval_script/eval_tools.zip
https://github.com/wondervictor/WiderFace-Evaluation
编译:
python3 setup.py build_ext --inplace
需要结果集:
def get_gt_boxes(gt_dir):
""" gt dir: (wider_face_val.mat, wider_easy_val.mat, wider_medium_val.mat, wider_hard_val.mat)"""
# gt_mat = h5py.File(os.path.join(gt_dir, 'wider_face_val.mat'), 'r')
gt_mat = loadmat(os.path.join(gt_dir, 'wider_face_val.mat'))
hard_mat = loadmat(os.path.join(gt_dir, 'wider_hard_val.mat'))
medium_mat = loadmat(os.path.join(gt_dir, 'wider_medium_val.mat'))
easy_mat = loadmat(os.path.join(gt_dir, 'wider_easy_val.mat'))