文件名称:*
文件大小:1.72MB
文件格式:ZIP
更新时间:2024-03-03 02:55:43
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)