首先需要在文件夹GlobalLocalImageCompletion_TF/data/img_align_celeba/ 放入图片数据,然后运行python train.py进行训练。
训练好这后,运行修复功能:python test.py --img_path=./data/test/388016.png
这个项目有一个缺点是进行图像修复时,只能把图像缩小再修复,修复出来的图像分辨率很差,因为他的模型把对应的tensor大小都固定了,后面可以考虑修改一下。
2.代码:https://github.com/pathak22/context-encoder
lua脚本实现,以及对应的tensorflow实现:
https://github.com/jazzsaxmafia/Inpainting
3.https://github.com/bamos/dcgan-completion.tensorflow