Yolo数据集划分（训练集+验证集+测试集）

一、前言

网上看了一些朋友写的代码，总感觉不太符合自己胃口，索性自己写一个，下面这个只需要提供图片所在文件夹和标签所在文件夹，就可以自动按比例划分训练集、验证集和测试集。

代码有比较详细的注释，如有疑问，请读者评论区留言或issue，我会尽快回复

二、代码

import os
import shutil
import random


def SplitDatasets(src_img_folder, src_label_folder, Dest_Folder, train_ratio, val_ratio, test_ratio):
    # 创建数据集存放文件夹
    Dest_train_img_folder, = os.path.join(Dest_Folder, 'train', 'images')
    Dest_train_label_folder = os.path.join(Dest_Folder, 'train', 'labels')
    Dest_valid_img_folder = os.path.join(Dest_Folder, 'valid', 'images')
    Dest_valid_label_folder = os.path.join(Dest_Folder, 'valid', 'labels')
    Dest_Test_img_folder = os.path.join(Dest_Folder, 'test', 'images')
    Dest_Test_label_folder = os.path.join(Dest_Folder, 'test', 'labels')
    Dir_List = [Dest_train_img_folder, Dest_train_label_folder,
                Dest_valid_img_folder, Dest_valid_label_folder,
                Dest_Test_img_folder, Dest_Test_label_folder]