[TensorFlow] ZeroPadding2D 레이어 예제
tf.keras.layers.ZeroPadding2D 레이어는 입력 데이터 주위에 제로 패딩을 추가하는 데 사용됩니다. 이를 통해 합성곱 레이어를 통과한 후의 특성 맵 크기를 조절할 수 있습니다. 아래는 tf.keras.layers.ZeroPadding2D 레이어를 사용한 예제입니다.
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import ZeroPadding2D, Conv2D
## 더미 이미지 데이터 생성
dummy_input = tf.random.normal(shape=(4, 10, 10, 3)) # 4개의 이미지, 각 이미지 크기: 10x10, 채널 개수: 3
## ZeroPadding2D 레이어 정의
padding_layer = ZeroPadding2D(padding=(1, 1)) # 상하좌우에 1씩 패딩 추가
## ZeroPadding2D 레이어 실행
padded_data = padding_layer(dummy_input)
print("Input shape:", dummy_input.shape)
print("Padded data shape:", padded_data.shape)`
위 예제에서는 tf.keras.layers.ZeroPadding2D 레이어를 사용하여 4개의 이미지에 대한 레이어를 정의하고 실행합니다. padding 매개변수로 상하 및 좌우에 추가할 패딩의 크기를 지정합니다.
실행 결과 예시:
Input shape: (4, 10, 10, 3)
Padded data shape: (4, 12, 12, 3)`
위에서 확인할 수 있듯이, 입력 데이터의 공간 차원이 2씩 확장되었습니다. tf.keras.layers.ZeroPadding2D 레이어를 사용하여 패딩을 추가하여 특성 맵의 크기를 조절할 수 있습니다.