[python] 폴더에서 원하는 파일 형태만 불러오기

특정 폴더에서 내가 원하는 파일 형태만 불러오기!

나같은 경우 특정 이미지 파일 형태만 불러와야할 때가 있다. 

예를 들어 .jpg, .png 등의 이미지 파일 형태가 섞인 폴더에서 .png 파일 형태만 불러오고 싶을 때!

 

 

import os 
import cv2

우선 필요한 라이브러리를 import 해오자. 

 

 

path = 'C:/Users/user/Desktop/folder/' # 폴더 경로
os.chdir(path) # 해당 폴더로 이동
files = os.listdir(path) # 해당 폴더에 있는 파일 이름을 리스트 형태로 받음

files 라는 리스트에는 다음과 같은 파일 이름이 담겨있다. 

리스트에 담긴 것은 어디까지나 '파일명'이기 때문에, 이미지를 이미지 파일로 불러오기 위해선 추가 작업이 필요하다. 

files 리스트 안의 파일명

 

 

png_img = []
jpg_img = []
for file in files:
    if '.png' in file: 
        f = cv2.imread(file)
        png_img.append(f)
    if '.jpg' in file: 
        f = cv2.imread(file)
        jpg_img.append(f)

cv2.imread('파일 이름')을 사용하여 이미지 파일을 불러올 수 있다. 

이렇게 실행해주면 png_img에는 .png 이미지 파일이, jpg_img에는 .jpg 이미지 파일이 numpy array 형태로 들어가게 된다. 

jpg_img

png_img