将图像按7:3划分train和val

问题:如何将原图按7:3划分train和val两个分类集呢?
代码:

# -*- coding:utf-8 -*-
#author :wu
# 将一个文件夹下图片按比例分在两个文件夹下,比例改0.7这个值即可
import os
import random
import shutil
from shutil import copy2
trainfiles = os.listdir('F:/dataset') #(图片文件夹)
num_train = len(trainfiles)
print( "num_train: " + str(num_train) )
index_list = list(range(num_train))
print(index_list)
random.shuffle(index_list)
num = 0
#trainDir = 'F:/dataset/train' #(将图片文件夹中的7份放在这个文件夹下)
trainDir = 'F:/train' #(将图片文件夹中的7份放在这个文件夹下)
if not os.path.exists(trainDir):
            os.makedirs(trainDir)
       
#validDir = 'F:/dataset/val'#(将图片文件夹中的3份放在这个文件夹下)
validDir = 'F:/val'#(将图片文件夹中的3份放在这个文件夹下)
if not os.path.exists(validDir):
            os.makedirs(validDir)
for i in index_list:
    fileName = os.path.join('F:/dataset', trainfiles[i])
    if num < num_train*0.7:
        print(str(fileName))
        copy2(fileName, trainDir)
    else:
        copy2(fileName, validDir)
    num += 1
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

会飞的渔WZH

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值