Python-subpixel利用Tensorflow的一个子像素卷积神经网络

subpixel：利用Tensorflow的一个子像素卷积神经网络

“““
Modification of https://github.com/stanfordnlp/treelstm/blob/master/scripts/download.py

Downloads the following:
- Celeb-A dataset
- LSUN dataset
- MNIST dataset
“““

from __future__ import print_function
import os
import sys
import gzip
import json
import shutil
import zipfile
import argparse
import subprocess
from six.moves import urllib

parser = argparse.ArgumentParser（description=‘Download dataset for DCGAN.‘）
parser.add_argument（‘--datasets‘ metavar=‘N‘ type=str nargs=‘+‘
                   help=‘name of dataset to download [celebA lusn mnist]‘）

def download（url dirpath）:
    filename = url.split（‘/‘）[-1]
    filepath = os.path.join（dirpath filename）
    u = urllib.request.urlopen（url）
    f = open（filepath ‘wb‘）
    filesize = int（u.headers[“Content-Length“]）
    print（“Downloading: %s Bytes: %s“ % （filename filesize））

    downloaded = 0
    block_sz = 8192
    status_width = 70
    while True:
        buf = u.read（block_sz）
        if not buf:
            print（‘‘）
            break
        else:
            print（‘‘ end=‘\r‘）
        downloaded += len（buf）
        f.write（buf）
        status = （（“[%-“ + str（status_width + 1） + “s] %3.2f%%“） %
            （‘=‘ * int（float（downloaded） / filesize * status_width） + ‘>‘ downloaded * 100. / filesize））
        print（status end=‘‘）
        sys.stdout.flush（）
    f.close（）
    return filepath

def unzip（filepath）:
    print（“Extracting: “ + filepath）
    dirpath = os.path.dirname（filepath）
    with zipfile.ZipFile（filepath） as zf:
        zf.extractall（dirpath）
    os.remove（filepath）

def download_celeb_a（dirpath）:
    NUM_EXAMPLES = 202599
    TRAIN_STOP = 162770
    VALID_STOP = 182637
    data_dir = ‘celebA‘
    if os.path.exists（os.path.join（dirpath data_dir））:
        print（‘Found Celeb-A - skip‘）
        return
    url = ‘https://www.dropbox.com/sh/8oqt9vytwxb3s4r/AADIKlz8PR9zr6Y20qbkunrba/Img/img_align_celeba.zip?dl=1&pv=1‘
    filepath = download（url dirpath）
    zip_dir = ‘‘
    with zipfile.ZipFile（filepath） as zf:
        zip_dir = zf.namelist（）[0]
        zf.extractall（dirpath）
    os.remove（filepath）

    # now split data into train/valid/test
    train_dir = os.path.join（dirpath zip_dir ‘train‘）
    valid_dir = os.path.join（dirpath zip_dir ‘valid‘）
    test_dir = os.path.join（dirpath zip_dir ‘test‘）
    if not os.path.exists（train_dir）:
        os.makedirs（train_dir）
    if not os.path.exists（valid_dir）:
        os.makedirs（valid_dir）
    if not os.path.exists（test_dir）:
        os.makedirs（test_dir）
    zip_path = os.path.join（dirpath zip_dir）
    for i in range（NUM_EXAMPLES）:
        image_filename = “{:06d}.jpg“.format（i+1）
        candidate_file = os.path.join（zip_path image_filename）
        if os.path.exists（candidate_file）:
            if i < TRAIN_STOP:
                dest_dir = train_dir
            elif i < VALID_STOP:
                dest_dir = valid_dir
            else:
                dest_dir = test_dir
            dest_file = os.path.j

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-09-26 17:12  subpixel-master\
     文件        1045  2017-09-26 17:12  subpixel-master\.gitignore
     文件        1078  2017-09-26 17:12  subpixel-master\LICENSE
     文件        8882  2017-09-26 17:12  subpixel-master\README.md
     目录           0  2017-09-26 17:12  subpixel-master\docker\
     文件        1303  2017-09-26 17:12  subpixel-master\docker\Dockerfile
     文件        5565  2017-09-26 17:12  subpixel-master\download.py
     目录           0  2017-09-26 17:12  subpixel-master\images\
     文件       29359  2017-09-26 17:12  subpixel-master\images\input_detail.png
     文件      437539  2017-09-26 17:12  subpixel-master\images\lowres_input.png
     文件      402972  2017-09-26 17:12  subpixel-master\images\more_examples.png
     文件       27698  2017-09-26 17:12  subpixel-master\images\no_padding_no_strides.gif
     文件      203725  2017-09-26 17:12  subpixel-master\images\padding_strides_transposed.gif
     文件       21285  2017-09-26 17:12  subpixel-master\images\ps_eq.png
     文件      363275  2017-09-26 17:12  subpixel-master\images\spcnn_diagram.png
     文件     1167559  2017-09-26 17:12  subpixel-master\images\superres_epoch0.png
     文件       69257  2017-09-26 17:12  subpixel-master\images\superres_epoch0_detail.png
     文件      919301  2017-09-26 17:12  subpixel-master\images\superres_epoch6.png
     文件       72255  2017-09-26 17:12  subpixel-master\images\superres_epoch6_detail.png
     文件        5624  2017-09-26 17:12  subpixel-master\keras_subpixel.py
     文件        2617  2017-09-26 17:12  subpixel-master\main.py
     文件        8009  2017-09-26 17:12  subpixel-master\model.py
     文件        5031  2017-09-26 17:12  subpixel-master\ops.py
     文件       56315  2017-09-26 17:12  subpixel-master\ponynet.ipynb
     文件        1410  2017-09-26 17:12  subpixel-master\subpixel.py
     文件        7007  2017-09-26 17:12  subpixel-master\utils.py