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目
2022-01-21 15:45:00
1.56MB
Python
1