YOLO 划分数据集（训练集、验证集、测试集）_如何将yolo数据集分开训练集验证集测试集

img_path = ‘./VOCdevkit/VOC2007/JPEGImages’# 你的图片存放的路径（路径一定是相对于你当前的这个脚本文件而言的）split_list = [0.8, 0.2]# 数据集划分比例[train:val]# Data是你要将要创建的文件夹路径（路径一定是相对于你当前的这个脚本而言的）训练集：验证集：测试集（7：2：1）标注文件是yolo格式（txt文件）

2401_84563987

1087人浏览 · 2024-05-07 15:32:23

2401_84563987 · 2024-05-07 15:32:23 发布

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

    train_label_dir = Data + '/labels/train'
    val_label_dir = Data + '/labels/val'
    # test\_label\_dir = Data + '/labels/test'

    # 创建文件夹
    os.makedirs(train_img_dir)
    os.makedirs(train_label_dir)
    os.makedirs(val_img_dir)
    os.makedirs(val_label_dir)
    # os.makedirs(test\_img\_dir)
    # os.makedirs(test\_label\_dir)

except:
    print('文件目录已存在')

train, val = split_list
all_img = os.listdir(img_path)
all_img_path = [os.path.join(img_path, img) for img in all_img]
# all\_label = os.listdir(label\_path)
# all\_label\_path = [os.path.join(label\_path, label) for label in all\_label]
train_img = random.sample(all_img_path, int(train \* len(all_img_path)))
train_img_copy = [os.path.join(train_img_dir, img.split('\\')[-1]) for img in train_img]
train_label = [toLabelPath(img, label_path) for img in train_img]
train_label_copy = [os.path.join(train_label_dir, label.split('\\')[-1]) for label in train_label]
for i in tqdm(range(len(train_img)), desc='train ', ncols=80, unit='img'):
    _copy(train_img[i], train_img_dir)
    _copy(train_label[i], train_label_dir)
    all_img_path.remove(train_img[i])
val_img = all_img_path
val_label = [toLabelPath(img, label_path) for img in val_img]
for i in tqdm(range(len(val_img)), desc='val ', ncols=80, unit='img'):
    _copy(val_img[i], val_img_dir)
    _copy(val_label[i], val_label_dir)

def _copy(from_path, to_path):
shutil.copy(from_path, to_path)

def toLabelPath(img_path, label_path):
img = img_path.split(‘\’)[-1]
label = img.split(‘.jpg’)[0] + ‘.txt’
return os.path.join(label_path, label)

if name == ‘__main__’:
img_path = ‘./VOCdevkit/VOC2007/JPEGImages’
label_path = ‘./YoloLabels’
split_list = [0.8, 0.2] # 数据集划分比例[train:val]
split_img(img_path, label_path, split_list)


## 训练集、验证集、测试集（7:2:1）


`split721.py`内容如下：

import os, shutil, random
from tqdm import tqdm

“”"
标注文件是yolo格式（txt文件）
训练集：验证集：测试集（7：2：1）
“”"

def split_img(img_path, label_path, split_list):
try:
Data = ‘./VOCdevkit/VOC2007/ImageSets’
# Data是你要将要创建的文件夹路径（路径一定是相对于你当前的这个脚本而言的）
# os.mkdir(Data)

    train_img_dir = Data + '/images/train'
    val_img_dir = Data + '/images/val'
    test_img_dir = Data + '/images/test'

    train_label_dir = Data + '/labels/train'
    val_label_dir = Data + '/labels/val'
    test_label_dir = Data + '/labels/test'

    # 创建文件夹
    os.makedirs(train_img_dir)
    os.makedirs(train_label_dir)
    os.makedirs(val_img_dir)
    os.makedirs(val_label_dir)
    os.makedirs(test_img_dir)
    os.makedirs(test_label_dir)

except:
    print('文件目录已存在')

train, val, test = split_list
all_img = os.listdir(img_path)
all_img_path = [os.path.join(img_path, img) for img in all_img]
# all\_label = os.listdir(label\_path)
# all\_label\_path = [os.path.join(label\_path, label) for label in all\_label]
train_img = random.sample(all_img_path, int(train \* len(all_img_path)))
train_img_copy = [os.path.join(train_img_dir, img.split('\\')[-1]) for img in train_img]
train_label = [toLabelPath(img, label_path) for img in train_img]
train_label_copy = [os.path.join(train_label_dir, label.split('\\')[-1]) for label in train_label]
for i in tqdm(range(len(train_img)), desc='train ', ncols=80, unit='img'):
    _copy(train_img[i], train_img_dir)
    _copy(train_label[i], train_label_dir)
    all_img_path.remove(train_img[i])
val_img = random.sample(all_img_path, int(val / (val + test) \* len(all_img_path)))
val_label = [toLabelPath(img, label_path) for img in val_img]
for i in tqdm(range(len(val_img)), desc='val ', ncols=80, unit='img'):
    _copy(val_img[i], val_img_dir)
    _copy(val_label[i], val_label_dir)
    all_img_path.remove(val_img[i])
test_img = all_img_path
test_label = [toLabelPath(img, label_path) for img in test_img]
for i in tqdm(range(len(test_img)), desc='test ', ncols=80, unit='img'):
    _copy(test_img[i], test_img_dir)
    _copy(test_label[i], test_label_dir)

def _copy(from_path, to_path):
shutil.copy(from_path, to_path)

def toLabelPath(img_path, label_path):
img = img_path.split(‘\’)[-1]
label = img.split(‘.jpg’)[0] + ‘.txt’
return os.path.join(label_path, label)

if name == ‘__main__’:
img_path = ‘./VOCdevkit/VOC2007/JPEGImages’ # 你的图片存放的路径（路径一定是相对于你当前的这个脚本文件而言的）