只需要face detection的功能,所以移除了Retinaface中特征点的分支,并进行灰度图输入训练,输入尺寸为128 * 128。
python train.py
10.26 已经修改使用 resnet18,尝试缩小主干网络,减少模型耗时的同时保证模型精度。
resnet网络修改
...
self.conv1_r = nn.Conv2d(1, self.inplanes, kernel_size=7, stride=2, padding=3,
bias=False)
...
11.11 retinaface对人脸的分类需要负样本的补充才能正常训练(必须要)