pytorch == 1.2.0
torchvision == 0.4.0
opencv-python == 4.4.0.44
Selected 4000 person (25 images for each) from UmdFaces dataset.
UmdFaces 提取码:yc9p
lfw 提取码:mm1v
lfw_test_pair 提取码:vjnu
1、修改train.py中parser的参数
2、python train.py
3、修改test.py中parser的参数
4、python test.py
目前可以调整的参数:
Backbone: num_layers=50/100/152 mode='ir'/'ir_se'
train: batch_size num_epoch lr stages weight_decay