VQVAE_image_inpainting

使用tensorflow 2.0 复现 VQVAE 图像修复

原论文地址：https://arxiv.org/abs/2103.10022

论文开源tensorflow 1.0 代码地址：https://github.com/USTC-JialunPeng/Diverse-Structure-Inpainting

运行环境：

• Ubunt 20.04
• tensorflow 2.3
• python 3.8

训练步骤

配置文件位于config文件夹中

修改配置文件 data_config.yaml 中的 train_flist 和 valid_flist 即可运行。

train_flist 和 valid_flist 的文件格式为：

/path/***.JPEG
/path/***.JPEG
/path/***.JPEG
/path/***.JPEG

训练依次过程如下：

python train_vqvae.py
python structure_generator.py
python texture_generator.py

留言

总算写完啦，小目标没达到。最近新项目比较忙，后续会陆续更新实验结果。

后续要有精力，会更新一个 image inpainting 的项目。 希望有大佬多加关注，多加指教

小弟恳请各位大佬可以多多指点，给予建议！若有些地方写的差的看不下去，也不要压制情绪，直接开喷吧！小弟受得住。