stylegan3-标签制作脚本

最新推荐文章于 2025-11-10 01:51:44 发布

原创

最新推荐文章于 2025-11-10 01:51:44 发布 · 320 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #pytorch

该脚本用于生成StyleGAN2-ADA-PyTorch的多类别JSON标签文件。它从输入文件夹读取数据，为每个文件创建标签，并将结果保存到指定的输出文件夹中的dataset.json。程序使用argparse模块处理命令行参数，os模块遍历文件系统，以及json模块进行序列化。


import argparse
import os
import random
import json

def parse_args():
	desc = "Tool to create multiclass json labels file for stylegan2-ada-pytorch" 
	parser = argparse.ArgumentParser(description=desc)

	parser.add_argument('--verbose', action='store_true',
		help='Print progress to console.')

	parser.add_argument('--input_folder', type=str,
		default=r'C:\Users\yewenjing\Desktop\GAN_input1',
		help='Directory path to the inputs folder. (default: %(default)s)')

	parser.add_argument('--output_folder', type=str,
		default=r'C:\Users\yewenjing\Desktop\GAN_input1',
		help='Directory path to the outputs folder. (default: %(default)s)')

	args = parser.parse_args()
	return args

def main():
	global args
	args = parse_args()
	base_dir = ''

	remakePath = args.output_folder
	if not os.path.exists(remakePath):
		os.makedirs(remakePath)

	data_dict = {}
	data_dict['labels'] = []
	label_counter = 0

	with open(os.path.join(remakePath, 'dataset.json'), 'w') a