1.使用前先在main.py
的目录下创建test_Imgs
文件夹,并将图片放在该文件夹下。
2.使用train_pnet.py
,train_rnet.py
,train_onet.py
对三个网络进行训练。并修改如下变量值。
save_path="saved_models/pnet.pt" #模型保存位置
dataset_path=r"path\to\data" #数据保存的位置
sum_path='./pnet_logs' #tensorboard数据保存位置
3.训练数据使用CelebA
数据集,并使用ImageCrop.py
生成,使用时需要对文件路径进行修改。
4.使用时执行main.py
即可
5.使用的结果