图片数据集读取

import os
import cv2
import numpy as np
from tqdm import tqdm

def read_image(path):
    '''读取路径下所有子文件夹中的图片并存入list'''
    img_list = []
    label_list = []
    dir_counter = 0
    
    for child_dir in os.listdir(path):
        child_path = os.path.join(path, child_dir)      
        for dir_image in tqdm(os.listdir(child_path)):
            img = cv2.imread(child_path+"\\"+dir_image,cv2.IMREAD_COLOR)
            # img =cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
            img_list.append(img)
            label_list.append(dir_counter)     
        dir_counter += 1
    return img_list,label_list,dir_counter