转自:https://blog.csdn.net/yunyi4367/article/details/80784205

参考:https://zhuanlan.zhihu.com/p/52188519

           https://github.com/willylulu/celeba-hq-modified

 

1.从Google drive上下载img_celaba.7z 
https://drive.google.com/drive/folders/0B7EVK8r0v71peklHb0pGdDl6R28 

也就是CelebA中的img_celeba.7z , 不是img_align_celeba_png.7z(文件是14个压缩包分卷,分别是001-014)

2.因为数据集是分卷压缩的,所以需要合并再解压缩

cat img_celeba.7z.0** > img_celeba.7z

把合并得到的img_celeba.7z解压缩得到img_celeba文件夹,内有202599张jpg图片. 

3.从Google drive上下载CelebA_hq_deltas 
https://drive.google.com/drive/folders/0B4qLcYyJmiz0TXY1NG02bzZVRGs 
文件夹内有30个zip文件,不需要解压缩. 
这两个资源如果没有用外网下载不了,可以自行查找百度云资源.不要找错了,不是img_align_celeba.zip. 


4.从celeba官网的下载地址或者别的地方下载list_landmarks_celeba.txt,很容易找到的。 


5.下载github上这位仁兄的h5tool.py,他的才能直接得到jpg文件,另外一个大兄弟的得到的是hdf5文件 
https://github.com/willylulu/celeba-hq-modified 
6.将h5tool.py,img_celeba,list_landmarks_celeba.txt和CelebA_hq_deltas文件夹放到同一目录下,然后创建save文件夹

mkdir celeba-hq
cd celeba-hq
mkdir celeba-64
mkdir celeba-128
mkdir celeba-256
mkdir celeba-512
mkdir celeba-1024


如果你只想要某一种尺寸的图,这个可以在代码里修改. 
6.修改h5tool.py里读取两个数据文件夹和txt文件的路径,使h5tool.py能正确读到文件 
7.运行h5tool.py,点开save文件夹,见证奇迹.

python2 h5tool.py create_celeba_hq 123456.h5 ./ ./

这样会有噪声


这样得到的高清图很多是有噪点的,后来发现在h5too.py的第400行左右有一个apply delta image的等式. 
取消添加delta image,也就是不令img = img + delta,而是直接img = img.这样就能得到没有噪点的高清图了. 
不明白为什么要加delta. 

不过亲测有效
 

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐