implement deepfill version 2 with paddlepaddle framework
- python3.6 above
- paddlepaddle 2.0
- opencv-python
Download pretrained weights of generator from following link()
链接:https://pan.baidu.com/s/1a352rmZaEaNk8PMaZObf0Q
提取码:sidy
-
train model
python train.py [--baseroot {path to image folder}] [--epoch {total epoch}]
-
resume training
python train.py --resume_epoch {num} [--baseroot {path to image folder}] [--epoch {total epoch}]
-
evaluate model
python eval.py [--load_path {path to generator state_dict}] [--baseroot {path to image folder} ]
-
inpaint images
python inpaint.py [--load_path {path to generator state_dict}] [--baseroot {path to image folder}] [--maskroot {mask to image folder}]
- well trained weight for datasets(need huge GPU resource for trainning)
- sketch