파이썬으로 CNN 구현하기: 이미지 데이터 분류

Table of Contents

안녕하세요, mj입니다!

오늘은 파이썬을 사용하여 CNN(합성곱 신경망)을 구현하는 방법에 대해 알아보겠습니다. CNN은 이미지 분류 및 인식 분야에서 매우 효과적인 딥러닝 모델입니다. 이번 포스팅을 통해 기본 개념과 함께 간단한 예제를 소개하겠습니다.

CNN의 기본 개념

CNN은 이미지와 같은 2D 데이터를 처리하기 위해 설계된 딥러닝 모델입니다. 기본적으로 합성곱 층, 풀링 층, 그리고 완전 연결 층으로 구성됩니다. 합성곱 층은 이미지에서 특징을 추출하고, 풀링 층은 차원을 축소하여 계산량을 줄입니다. 마지막으로 완전 연결 층은 최종 분류를 수행합니다.

필요한 라이브러리 설치

먼저, 필요한 라이브러리를 설치해야 합니다. 아래의 코드를 사용하여 TensorFlow와 Keras를 설치하세요.

pip install tensorflow

데이터셋 준비

이번 예제에서는 CIFAR-10 데이터셋을 사용하겠습니다. 이 데이터셋은 10개의 클래스(비행기, 자동차, 새 등)로 구성된 60,000개의 이미지로 이루어져 있습니다. TensorFlow를 사용하여 쉽게 불러올 수 있습니다.


from tensorflow.keras.datasets import cifar10

(X_train, y_train), (X_test, y_test) = cifar10.load_data()
X_train, X_test = X_train / 255.0, X_test / 255.0

CNN 모델 구축

이제 CNN 모델을 구축해 보겠습니다. Keras의 Sequential API를 사용하여 간단하게 모델을 만들 수 있습니다.


from tensorflow.keras import layers, models

model = models.Sequential([
    layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)),
    layers.MaxPooling2D((2, 2)),
    layers.Conv2D(64, (3, 3), activation='relu'),
    layers.MaxPooling2D((2, 2)),
    layers.Conv2D(64, (3, 3), activation='relu'),
    layers.Flatten(),
    layers.Dense(64, activation='relu'),
    layers.Dense(10, activation='softmax')
])

모델 컴파일 및 훈련

모델을 컴파일하고 훈련하는 단계입니다. 손실 함수와 옵티마이저를 설정하고, 훈련을 시작합니다.


model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(X_train, y_train, epochs=10, validation_data=(X_test, y_test))

모델 평가 및 예측

훈련이 완료된 후, 테스트 데이터셋을 이용해 모델의 성능을 평가합니다. 또한, 몇 가지 이미지를 예측해 보겠습니다.


test_loss, test_acc = model.evaluate(X_test, y_test, verbose=2)
print('\n테스트 정확도:', test_acc)

predictions = model.predict(X_test)
print(predictions[0])  # 첫 번째 이미지에 대한 예측