如何从文件夹中批量读取图像,并划分数据集
返回的是列表
import os
import tqdm
from tqdm import tqdm #显示进度
from imageio import imread
from sklearn.model_selection import train_test_split
x=[]
y=[]
def INPUT_IMAGE(TRAIN_IMG_PATH,TRAIN_MASK_PATH):
for filename in tqdm(os.listdir(r"./"+TRAIN_IMG_PATH),desc="训练图像加载进度",position=0):
img =imread(TRAIN_IMG_PATH + "/"+filename)
x.append(img)
for filename in tqdm(os.listdir(r"./"+TRAIN_MASK_PATH),desc="训练掩码加载进度",position=0):
mask=imread(TRAIN_MASK_PATH+"/"+filename)
y.append(mask)
return x,y
TRAIN_IMG_PATH="mini_train_img"
TRAIN_MASK_PATH="mini_train_mask"
X,Y=INPUT_IMAGE(TRAIN_IMG_PATH,TRAIN_MASK_PATH)
X_train, Y_train, X_valid, Y_valid = train_test_split(X, Y, test_size = 0.3, random_state=2020)