DCGAN之download.py

  • 首先须要导入的包中,gzipzipfile用于文件压缩和解压缩相关;argparse用于构建命令行参数;requests用于http请求下载网络文件资源;subprocess用于运行shell命令;tqdm用于进度条显示;six包用于python2和python3的兼容,好比 from six.moves import urllib 这句就是导入python2.x的urllib库。
  • 上面的代码除了原做者加的注释以外,我也已经加了一部分注释,意思应该比较好理解了。主要作的事情,就是利用requests库从网络上将mnist,lsun以及celebA这三个数据集下载下来,保存在data目录下。注意mnistcelebA数据集下载下来以后还进行了解压缩。
  • 上面的三个数据集,mnist是著名的手写数字数据库,你们应该都已经很熟悉了;lsun是大型场景理解数据集(large-scale-scene-understanding);celebA是一个开源的人脸数据库。除了mnist以外,其他两个数据集体积都较大,celebA大概有20w+的图像,压缩文件体积为1.4G;而lsun有不少个场景不一样的数据集,若是按照上面的脚本下载,下载的文件为bedroom数据集,压缩文件有46G之大,并且其实下载下来的文件解压后为mdb(Access数据库)格式,不是原始图片格式,不方便处理。因此咱们实际会下载其余的数据集做为替代,好比这个room layout estimation(2G)数据。若是使用download.py脚本下载速度较慢的话,能够自行下载好数据集,而后放在data目录下便可。

    DCGAN 代码简单解读
    http://www.mamicode.com/info-detail-2312795.htmlhtml

相关文章
相关标签/搜索