1:背景介绍
为了训练网络,采集数据集若干组,每组文件下面都有img和label两个子文件,分别存着图片和txt。如图
现在想把他们存到一起
2:程序
# coding=utf-8
import os
import shutil
def extractdoc(name):
#目标文件夹
root_dir ='./'
pattern = '1111'
set_list = '11'
determination = root_dir + "/" + pattern + set_list + "/"+name
if not os.path.exists(determination):
os.makedirs(determination)
#源文件夹路径
path = '222'
folders = os.listdir(path)
print (folders)
for folder in folders:
dir = path + '/' + str(folder)+'/'+name
files = os.listdir(dir)
for file in files:
source = dir + '/' + str(file)
deter = determination + '/' + str(file)
shutil.copyfile(source, deter)
extractdoc('img')
extractdoc('label')