【分类数据集中训练集和验证集划分(无标签)】

分类数据集中训练集和验证集划分(无标签)

import os
import shutil
import random
from tqdm import tqdm 


img_path = './rect_tunnel_v0504/rect_tunnel_soda_0508_0/'
train_save_path = './rect_tunnel_v0504/train/rect_tunnel_train_0508_0/'
val_save_path = './rect_tunnel_v0504/val/rect_tunnel_test_0508_0/'

if not os.path.exists(train_save_path):
    os.makedirs(train_save_path)
if not os.path.exists(val_save_path):
    os.makedirs(val_save_path)

imgs = os.listdir(img_path)
random.shuffle(imgs)
random.shuffle(imgs)
rate = 0.92 # train、val划分比例
for i, img in enumerate(tqdm(imgs)):
    if i< len(imgs)* rate:
        shutil.copy(img_path+ img, train_save_path+ img)
    else:
        shutil.copy(img_path+ img, val_save_path+ img)