遍历ImageNet或者Place365_256x256或者Celeba生成flist

了悟生死大事

已于 2022-02-16 23:24:51 修改

阅读量562

点赞数

分类专栏： python tensorflow 文章标签： p2p cnn 网络协议

于 2022-02-16 23:24:00 首次发布

本文链接：https://blog.csdn.net/nil_lu/article/details/122974016

版权

该博客介绍了一个Python脚本，用于遍历Place365_256x256或类似数据集的文件夹结构，生成训练和验证用的flist文件。脚本支持对文件路径进行随机打乱，并将结果写入到指定的flist文件中。

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

#for windows
import argparse
import os
from random import shuffle

parser = argparse.ArgumentParser()
parser.add_argument('--folder_path', default='./place365_256x256', type=str,
help='The folder path')
parser.add_argument('--train_filename', default='./place_data_flist/train_shuffled.flist', type=str,
help='The train filename.')
parser.add_argument('--validation_filename', default='./place_data_flist/validation_shuffled.flist', type=str,
help='The validation filename.')
parser.add_argument('--is_shuffled', default='1', type=int,
help='Needed to be shuffled')

if __name__ == "__main__":

args = parser.parse_args()

# make 2 lists to