colab

os.listdir('folder_path')

출력: file list로 출력됨

# 예시
import os
import matplotlib.image as img
import numpy as np

train_forest_path = '/content/mountainForest/train/forest/'

train_forest_imgs = []
for i in os.listdir(train_forest_path):
    if 'jpg' in i:
        train_forest_imgs.append(img.imread(train_forest_path + i))
        
train_forest_imgs = np.array(train_forest_imgs)

local

glob.glob('folder_path' + '*.jpg')	# jpg 파일만 찾아서 list

# 예시
train_forest_path = './mountainForest/mountainForest/train/forest/'
train_forest_imgs = []


for i in glob.glob(train_forest_path + '*.jpg'):
    train_forest_imgs.append(img.imread(i))

train_forest_imgs = np.array(train_forest_imgs)