缺点:文件夹分的太散,合并dat会丢到很多文件。
data_path = r'mao/'
target_path = r'0724/'
dirs=[]
for dirpath, dirnames, filenames in os.walk(data_path):
if 'JPEGImages' in dirnames:
dirs.append(dirpath.replace(data_path,"/"))
os.makedirs(target_path, exist_ok=True)
processes = []
for dir in dirs:
target_dir =target_path + dir
os.makedirs(target_dir, exist_ok=True)
data_dir = data_path + dir
print(data_dir)
p = mp.Process(target=gen_data, args=(batch_size, data_dir, target_path, img_size))
p.start()
processes.append(p)
for p in processes:
p.join()