上传者: 42108054
|
上传时间: 2022-01-21 15:45:00
|
文件大小: 1.56MB
|
文件类型: -
FCHD
本工程是官方工程的python3版本,重构了部分代码,并训练得到了一个效果不错的模型。
论文地址:
官方工程地址:
安装
Clone
git clone https://github.com/embracesource-cv-com/FCHD.git
安装依赖
cd FCHD
pip install -r requirements.txt
数据配置
在下载Brainwash数据集,caffe预训练模型,以及人头检测模型
将预训练模型vgg16_caffe.pth放到./checkpoints/pre_trained目录下
将检测模型checkpoint_best.pth放到./checkpoints目录下
将brainwash数据集放到/home/dataset/fchd_datas目录下
当然你可以自定义各个路径,然后在config.py中修改相应的配置
预测
在工程root目