实现照片补全 with GAN IN MATLAB

机器学习期末项目
根据paper
a.Painting Outside the Box: Image Outpainting with GANs
b.Globally and Locally Consistent Image Completion

1.dataset
place365
可在第一篇paper中找到
也可在此下载
val_365

2.model ----GAN
实现照片补全 with GAN IN MATLAB_第1张图片
用了两个CNN,一个discriminator,一个generator
不懂GAN 的可以自行了解一下

3.loss
实际mse挺低但边缘还是糊,原paper 用了两个local discriminator
和一个global discriminator,并且有一步postprocessing 处理,我们在matlab 中只用了一个discriminator,所以loss率低,但其实效果还不是很好
实现照片补全 with GAN IN MATLAB_第2张图片在家做自己电脑不行,让朋友帮忙train 他发过来的一部分loss

4.结果如下:

实现照片补全 with GAN IN MATLAB_第3张图片
实现照片补全 with GAN IN MATLAB_第4张图片
实现照片补全 with GAN IN MATLAB_第5张图片
建议用MATLAB2020a,其他版本train过程可能有版本兼容问题
code with matlab

你可能感兴趣的:(机器学习with,matlab)