CelebA-HQ的jpg格式版转化

最新推荐文章于 2024-09-20 20:09:48 发布

转载最新推荐文章于 2024-09-20 20:09:48 发布 · 686 阅读

深度学习专栏收录该内容

164 篇文章

订阅专栏

本文详细介绍了如何从Google Drive下载并处理CelebA-HQ数据集，包括合并与解压分卷压缩文件、下载辅助工具与资源、调整代码参数以生成无噪点高清图像，并解决图片命名与数量问题。

1.从Google drive上下载img_celaba.7z
https://drive.google.com/drive/folders/0B7EVK8r0v71peklHb0pGdDl6R28
2.因为数据集是分卷压缩的,所以需要合并再解压缩

cat img_celeba.7z.0** > img_celeba.7z
1
把合并得到的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
1
2
3
4
5
6
7
如果你只想要某一种尺寸的图,这个可以在代码里修改.
6.修改h5tool.py里读取两个数据文件夹和txt文件的路径,使h5tool.py能正确读到文件
7.运行h5tool.py,点开save文件夹,见证奇迹.

python h5tool.py create_celeba_hq 123456.h5 ./ ./
1
这样得到的高清图很多是有噪点的,后来我发现在h5too.py的第400行左右有一个apply delta image的等式.
我取消添加delta image,也就是不令img = img + delta,而是直接img = img.这样就能得到没有噪点的高清图了.
不明白为什么要加delta.
最后,如果不修改代码的话生成的图片名称和原低清图不一致,这个很容易改,直接将process_func函数返回的idx改成orig_file就好了.另外,作者生成的图片是从索引7000多开始的,这样最后会少7000多张图.估计是他生成时中断过忘记改回从0开始了.
---------------------
作者：祖国的花朵33
来源：优快云
原文：https://blog.youkuaiyun.com/yunyi4367/article/details/80784205?utm_source=copy
版权声明：本文为博主原创文章，转载请附上博文链接！