CNN 모델의 사용자 이미지 데이터 생성하기

인물이나 동물 등 물체사진이나 위성사진을 이용해서 컨볼루션 뉴럴 네트워크인 CNN모델을 구현하고 물체를 인식하고 분류하기 위해서는 모델에 맞도록 입력 데이터를 생성하여야 한다.

이를 위해서 넘파일 배열자료인 npy포맷으로 파일을 생성하는 것이 하나의 방법일 수 있다.

예를 들어 이미지 폴더가 c:/image_folder이고 tif 이미지 파일이 16000장이 있다고 가정할 때 입력데이터는 glob 함수와 cv2 패키지의 imread함수를 이용하여 다음 구문처럼 생성될 수 있다.

1. 패키지 로드하기

import numpy as np
import cv2
import glob

2. 입력데이터 생성하기

train_data=[]

for img in glob.glob("c:/image_folder/*.tif"):
    image = cv2.imread(img, 1)
    train_data.append(image)

train_data=np.array(train_data)

train_data.shape