目的:顺序读取并加载文件夹的所有文件或图片
描述:图1显示的路径是根目录,我需要通过python读取并加载根目录下文件夹中所有文件或图片,图2,3,4,5分别为根目录下文件夹中的图片
图1
图2
图3
图4
图5
python代码:
DATA_DIR = os.getcwd() + "/Images/"
RESIZE_TO = 100, 100
x, y = [], []
for i in range(len(os.listdir(DATA_DIR))):
for path in [f for f in os.listdir(DATA_DIR+os.listdir(DATA_DIR)[i])]:
x.append(cv2.resize(cv2.imread(DATA_DIR + os.listdir(DATA_DIR)[i] + '/' + path), (RESIZE_TO)))
label = os.listdir(DATA_DIR)[i]
y.append(label)
x, y = np.array(x), np.array(y)
y_label = y
print(x,y_label )
代码结果: