- 下载WIDER_FACE数据集
如有过期,我会不定时更新
- 下载数据集标签文件,https://github.com/sovrasov/wider-face-pascal-voc-annotations
- 将WIDER_train_annotations 和 WIDER_val_annotations文件名更改为Annotations,并分别放置在WIDER_train 和 WIDER_val两个文件下,并将改名前WIDER_train_annotations 和 WIDER_val_annotations两个文件夹中的train.txt和val.txt分别移动至data/WIDERFace文件下。
整理后的文件如下形式:
mmdetection
├── mmdet
├── tools
├── configs
├── data
│ ├── WIDERFace
│ │ ├── WIDER_train
│ | │ ├──0--Parade
│ | │ ├── ...
│ | │ ├── Annotations
│ │ ├── WIDER_val
│ | │ ├──0--Parade
│ | │ ├── ...
│ | │ ├── Annotations
│ │ ├── val.txt
│ │ ├── train.txt
- 修改…/base/datasets/wider_face.py文件中,WIDER_FACE数据集的加载路径
- mmdetection中只提供了一个SSD300的结构,也可以自己写一个,选择合适的网络结构开始训练