【文件属性】:
文件名称:STDN-PyTorch:这是 PyTorch 中“On Disentangling Spoof Traces for Generic Face Anti-Spoofing”的非官方实现
文件大小:33KB
文件格式:ZIP
更新时间:2021-05-29 10:54:24
STDN-PyTorch
这是 PyTorch 中“On Disentangling Spoof Traces for Generic Face Anti-Spoofing”的非官方实现
[] [ ]
要求
Python 3.6+
PyTorch 1.6.0
易字典
TensorFlow 2+
资料准备
使用以下任何存储库为实时面部和欺骗面部创建关键点:
创建一个列名称为“rgb_path”、“keypoints”、“label”的 csv
对于标签,欺骗是1,实数是0
对数据集进行同等采样,使欺骗的数量和真实的数量相等。
配置
要更改您的配置,请打开config.py并设置数据路径,请更改flags.data_config 。 您也可以根据需要更改其他参数。
如何训练?
完成数据准备并创建数据路径后,运行以下代码:
python train.py
您的所有检查点都将保存
【文件预览】:
STDN-PyTorch-main
----loss.py(12KB)
----train.py(314B)
----dataloader.py(2KB)
----LICENSE(1KB)
----trainer.py(12KB)
----model.py(17KB)
----config.py(7KB)
----README.md(2KB)
----utils()
--------transforms.py(8KB)
--------__pycache__()
--------joint_transforms.py(6KB)
--------scripts.py(3KB)
--------__init__.py(0B)
--------samplers.py(3KB)
--------config_reader.py(4KB)