Python脚本——将一个文件夹下的文件划分到不同文件夹_python 将文件夹里的项目根据各自的名称组成一个以项目为名称独立的文件夹

# 目标检测中，将pascal格式的数据标签Annotations按照train.txt、val.txt分成对应的文件夹
# 用在retinanet中的训练集和验证集

import os
import shutil
 
def split_file(src_file_path, dst_file, dst_file_path):
    file = open(dst_file, 'r').readlines()
    for line in file:
        line = line.strip('\n')
        src_file_name = src_file_path + '{}.xml'.format(line)
 
        if not os.path.isfile(src_file_name):
            print("%s not exist !!!"%(src_file_name))
        else:
            dst_file_name = dst_file_path + '{}.xml'.format(line)
            shutil.copyfile(src_file_name,dst_file_name)
            print("copy %s -->> %s"%(src_file_name, dst_file_name))
 
 
if __name__ == '__main__':
    src_path = '/data/Annotations/'
    train_path = '/data/train_set/'
    val_path = '/data/val_set/'
    train_file = '/data/ImageSets/Main/train.txt'
    val_file = '/data/ImageSets/Main/val.txt'
    split_file(src_path, train_file, train_path)
    split_file(src_path, val_file, val_path)