python 按比例分割数据集

最新推荐文章于 2024-10-30 19:17:19 发布

原创最新推荐文章于 2024-10-30 19:17:19 发布 · 2.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python

烂笔头专栏收录该内容

37 篇文章

订阅专栏

本文介绍了如何使用Python脚本将图片数据集划分为训练集和验证集，采用Yolov5训练框架，通过随机打乱并按比例切割实现了80%样本用于训练，20%用于验证的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用于yolov5训练。

import os
import random

images_path = '/data/my_set/images'
train_percent = 0.8

total_images = os.listdir(images_path)
random.shuffle(total_images) #打乱列表顺序

num = len(total_images)
carve_num = int(num*train_percent)
train = total_images[:carve_num]
val = total_images[carve_num:]

with open('train.txt', 'w') as f:
    for train_txt in train:
        f.write(os.path.join(images_path,train_txt) + '\n')
with open('val.txt', 'w') as f:
    for val_txt in val:
        f.write(os.path.join(images_path,val_txt) + '\n')