【文件属性】:
文件名称:*
文件大小:1.72MB
文件格式:ZIP
更新时间:2021-02-11 09:09:03
Python
中心损耗
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
【文件预览】:
Center-master
----models()
--------resnet.py(2KB)
--------__init__.py(57B)
--------base.py(465B)
----loss.py(1KB)
----main.py(10KB)
----images()
--------softmax.png(113KB)
--------trump_a.png(392KB)
--------obama_b.png(375KB)
--------roc.png(11KB)
--------result.jpg(133KB)
--------obama_a.png(384KB)
--------result.png(133KB)
--------center-loss.png(83KB)
--------trump_b.png(140KB)
----utils.py(1KB)
----metrics.py(2KB)
----dataset.py(9KB)
----tests()
--------__init__.py(317B)
--------center_loss_test.py(2KB)
----trainer.py(6KB)
----imageaug.py(613B)
----.gitignore(24B)
----device.py(69B)
----README.md(4KB)
----pick.py(397B)