inception_score.py 文件源码

def load_images(o_dir, i_dir, n_images=3000, size=128):
    prepare_inception_data(o_dir, i_dir)
    image_list = []
    done = False
    cnt = 0
    bar = progressbar.ProgressBar(redirect_stdout=True,
                                  max_value=progressbar.UnknownLength)
    for root, dirs, files in os.walk(o_dir):
        if files:
            for f in files:
                cnt += 1
                file_dir = os.path.join(root, f)
                image_list.append(ip.load_image_inception(file_dir, 0))
                bar.update(cnt)
                if len(image_list) == n_images:
                    done = True
                    break
        if done:
            break
    bar.finish()
    print('Finished Loading Files')
    return image_list