BEGAN-carpedm代码调试

最新推荐文章于 2021-10-28 11:04:14 发布

羽扇纶巾v

最新推荐文章于 2021-10-28 11:04:14 发布

阅读量139

点赞数

分类专栏： GAN调试

本文链接：https://blog.csdn.net/qq_42978803/article/details/116305399

版权

GAN调试专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用Python脚本来管理和组织 CelebA 数据集。作者首先将数据集的图片放在'images'文件夹中，然后通过代码创建了训练集、验证集和测试集的目录，并对数据进行了相应的划分。接着，通过建立软链接的方式，将原始图片链接到对应的训练、验证和测试目录下，以便于后续的训练过程。当训练中断后，可以通过指定加载路径继续训练。

摘要由CSDN通过智能技术生成

总共数据是202599张图片，在这里的分了训练集162770（比例0.8），验证集19867（比例0.098），测试集19962（比例0.096）

注意文件夹的创建

##base_path = './data'
def add_splits(base_path):
    data_path = os.path.join(base_path, 'CelebA')
    images_path = os.path.join(data_path, 'images')
    train_dir = os.path.join(data_path, 'splits', 'train')
    valid_dir = os.path.join(data_path, 'splits', 'valid')
    test_dir = os.path.join(data_path, 'splits', '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)

这里不用给的download.py来下载CelebA数据集，我自己已经准备好了，把CelebA数据集里的图片全放在images文件夹里。

if __name__ == '__main__':
    base_path = './data'
    prepare_data_dir()
    ##download_celeb_a(base_path)
    add_splits(base_path)

把download_celeb_a(base_path)这行注释掉，再以管理员权限运行download.py

def check_link(in_dir, basename, out_dir):
    in_file = os.path.join(in_dir, basename)
    if os.path.exists(in_file):
        link_file = os.path.join(out_dir, basename)
        rel_link = os.path.relpath(in_file, out_dir)
        os.symlink(rel_link, link_file)