当前位置:   article > 正文

制作CelebA-HQ数据集(所有文件全在百度网盘,永久可用)

celeba-hq

目录

1、下载img_celeba.7z

2、下载 list_landmarks_celeba.txt

3、下载celeba-hq-deltas文件夹

4、下载python文件

5、在命令行里面运行程序生成图像


整个文件夹的目录如下:

CelebA

        img_celeba.7z

        img_celeba

                img_celeba

                list_landmarks_celeba.txt

        celeba-hq-deltas

        CelebA-HQ

                celeba-64

                celeba-128

                celeba-256

                celeba-512

                celeba-1024

1、下载img_celeba.7z

链接: https://pan.baidu.com/s/1GfcwgzAr2HJYH68nLqDssg 提取码: 7ryr 。下载完后全部解压到一个文件夹中。

2、下载 list_landmarks_celeba.txt

链接: https://pan.baidu.com/s/13l-Z8exetF4p7gkoPqMYxQ 提取码: ckn5。下载好后把此txt文件放入img_celeba文件夹中。

3、下载celeba-hq-deltas文件夹

链接: https://pan.baidu.com/s/1Gb7nDHcOnuoWGukfO5fsKA 提取码: 4hcx 。下载完后不用管了,压缩包不要解压!!

4、下载python文件

 首先,在anaconda创建一个python是2.7的虚拟环境

conda create -n python2.7 python=2.7

下载依赖包

  1. # 激活虚拟环境
  2. conda activate python2.7
  3. pip install numpy
  4. pip install scipy
  5. pip install pillow
  6. pip install h5py
  7. pip install cryptography

 下载生成图像的python文件

链接: https://pan.baidu.com/s/1y65nWRylR6OMOvNL2W3jtw 提取码: iavg 。

 对里面的一些代码进行修改:1、修改list_landmarks_celeba.txt文件的读取地址;2、修改生成的HQ图像的保存地址(celeba-64这些几个文件夹需要在运行代码前创建好,程序不会自动创建)。

  1. # 把中间的Anno删掉
  2. with open(os.path.join(celeba_dir, 'list_landmarks_celeba.txt'), 'rt') as file:
  3. # 修改生成的celeba-hq图像的保存地址
  4. # Save all generated images.
  5. root_path = '~/Project/Datasets/CelebA/CelebA-HQ'
  6. with ThreadPool(num_threads) as pool:
  7. for orig_fn, aimg64, aimg128, aimg256, aimg512, aimg1024 in pool.process_items_concurrently(fields['idx'], process_func=process_func, max_items_in_flight=num_tasks):
  8. aimg64.save(root_path+'/celeba-64/'+str(orig_fn))
  9. aimg128.save(root_path+'/celeba-128/'+str(orig_fn))
  10. aimg256.save(root_path+'/celeba-256/'+str(orig_fn))
  11. aimg512.save(root_path+'/celeba-512/'+str(orig_fn))
  12. aimg1024.save(root_path+'/celeba-1024/'+str(orig_fn))
  13. print(orig_fn)

5、在命令行里面运行程序生成图像

  1. cd ~/celeba-hq-modified-master
  2. python h5tool.py create_celeba_hq celeba-hq-1024x1024.h5 ~/CelebA/img_celeba ~/CelebA/celeba-hq-deltas

程序要运行很久,放着就行。 

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/238574
推荐阅读
相关标签
  

闽ICP备14008679号