YOLO数据集划分

这段代码用于将YOLO数据集按照指定的比例划分为训练集和验证集,并分别保存图像和对应的标签文件。通过随机种子确保每次划分一致,设置训练集比例为0.9,最终打印出图片总数、训练集数量和验证集数量。
摘要由CSDN通过智能技术生成

import os
import cv2
import random
from PIL import Image
# os.makedirs('labels_train')
# os.makedirs('labels_val')

img_train_path ='.//images/train/'  #图片训练集保存路径,需要自己新建
img_val_path = './/images/val/'     #图片验证集保存路径   需要自己新建
labels_train_path = './custom/labels/train/'       #标签训练集保存路径   需要自己新建
labels_val_path = './custom/labels/val/'           #标签验证集保存路径   需要自己新建
dir_images = './img/'                       #数据集路径
dir_labels  = './ann/'                  #标签路径

images = os.listdir(dir_images)             #把所有的图片名放入一个列表中
labels = os.listdir(dir_labels)            #labels的顺序和images的顺序不一定是一致的

random.seed(2021)                           #设置一个随机种子,确保每次运行都按照既定的随机形式
random.shuffle(images)                      #洗牌操作,打乱列表顺序

train_spilt_rate =0.9       #数据集划分比例,数据集较小

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值