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
图片数据集读取
猜你喜欢
转载自blog.csdn.net/changyan_123/article/details/83386711
今日推荐
周排行