因为我的文件夹结构是
fourthdown/CVPR/ababab/train_sets
——1
——1
——2
……
——n
——2
……
——n
然后该代码的功能是,移动图片,使得每七张图片存在一个新的文件夹中,移动时,如果被处理的文件夹中当前图像/文件有后六个文件,才进行移动,否则进入下一个文件夹。
举个例子,一个文件夹中有10个图片/文件,那么1-7,2-8,3-9,4-10,这四组各七张图片/文件存入新的文件夹中,命名为1、2、3、4,然后该文件夹处理完毕,继续处理下一个文件夹。
import os
import shutil
def moveFile(fileDir,tarDir):
if not os.path.isdir(fileDir): #查看数据集是否存在
print('Error: No source data found')
exit(0)
if not os.path.isdir(tarDir): #查看存储文件夹是否存在,不存在则创建,train
os.mkdir(tarDir)
paths = os.listdir(fileDir)
for path in paths:
source_dir = fileDir + '/' + path #H:/fourthdown/CVPR/ababab/train_sets/1
path_1 = tarDir + '/' + path #H:/fourthdown/CVPR/ababab/train/1
if not os.path.isdir(os.path.join(tarDir, path)): #/train/1
os.mkdir(os.path.join(tarDir, path))
count = 1
dirs = os.listdir(source_dir)
for dir in dirs:
dir_path = source_dir + '/' +dir #H:/fourthdown/CVPR/ababab/train_sets/1/1
file_list = os.listdir(dir_path)
num_classes = len(file_list) #/1/1图片数目
for i in range(num_classes): #遍历/1/1所有图片
if ((i+6) < num_classes): #当前图片有后七张图片
path_2 = path_1 + '/' + str(count) # H:/fourthdown/CVPR/ababab/train/1/1
if not os.path.isdir(path_2):
os.mkdir(path_2)
count = int(count) + 1
for n in range(1,7):
class_name = file_list[i]
save_name = 'im' + str(n) + '.png'
image_dir = dir_path + '/' + class_name
path_3 = path_2 + '/' + save_name
shutil.copyfile(image_dir,path_3)
if __name__=='__main__':
fileDir = "H:/fourthdown/CVPR/ababab/test_sets" #原始图片文件夹路径
tarDir = "H:/fourthdown/CVPR/ababab/test" #移动到新文件夹的路径
moveFile(fileDir,tarDir)