tensorflow图像修复代码解析

详见代码: https://github.com/shinseung428/GlobalLocalImageCompletion_TF

首先需要在文件夹GlobalLocalImageCompletion_TF/data/img_align_celeba/ 放入图片数据,然后运行python train.py进行训练。

训练好这后,运行修复功能:python test.py --img_path=./data/test/388016.png

这个项目有一个缺点是进行图像修复时,只能把图像缩小再修复,修复出来的图像分辨率很差,因为他的模型把对应的tensor大小都固定了,后面可以考虑修改一下。

猜你喜欢

转载自blog.csdn.net/bvngh3247/article/details/88389100