根据Label标签将数据集划分

一、标签分为train、val、test三个，格式为YOLO的txt，根据txt名称将images的图片进行分类划分。

label样式

 二、image样式，和label在同一级目录，里面为未分类的所有图片。 

 

 三、划分代码 仅需要改动图片路径和label路径即可

import os
import shutil
from pathlib import Path
file_type = ['train', 'val', 'test']
base_path = '../label/'      #标签train等文件的父目录
images = '../images/'        #总的图像文件路径
for tp in file_type:
    annotation_path = base_path + tp
    file_list = os.listdir(annotation_path)
    Path(base_path + tp + 'images').mkdir(parents = True)  # 参数 parents 表示如果父目录不存在，是否要创建父目录
    for fl in file_list:
        fl = fl[:-3] + 'png'
        shutil.move(os.path.join(images, fl), os.path.join(base_path + tp + 'images', fl))

四、效果 已经成功划分数据集