中心损耗
pytorch中人脸识别的中心损失实现。 论文发表于:
要求
Python 3.6
火炬0.4
用法
培训无需下载任何内容。 以下命令将创建目录并自动下载所有内容。
# For quick reference and small GPU ram
python3 main.py --arch resnet18 --batch_size 64 --epochs 50
# For a more solid model
python3 main.py
# or
python3 main.py --arch resnet50 --batch_size 256 --epochs 150
评价
python3 main.py --evaluate ./logs/models/epoch_xx.pth.tar
# Model accuracy is 0.961722195148468
# R