Python从文件夹中批量读取文件（显示进度）

如何从文件夹中批量读取图像，并划分数据集
返回的是列表

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)