핸즈온 머신러닝(2판): 사이킷런, 케라스, 텐서플로 2를 활용한 머신러닝, 딥러닝 완벽 실무

577

14

장

합성곱 신경망을 사용한 컴퓨터 비전

model.add(keras.layers.BatchNormalization())

model.add(keras.layers.Activation("relu"))

model.add(keras.layers.MaxPool2D(pool_size=3, strides=2, padding="same"))

prev_filters = 64

for filters in [64] * 3 + [128] * 4 + [256] * 6 + [512] * 3:

strides = 1 if filters == prev_filters else 2

model.add(ResidualUnit(filters, strides=strides))

prev_filters = filters

model

.add(keras.layers.GlobalAvgPool2D())

model.add(keras.layers.Flatten())

model.add(keras.layers.Dense(10, activation="softmax"))

이 코드에서 조금 복잡한 부분은 모델에

ResidualUnit

층을 더하는 반복 루프입니다. 앞서 설

명한 것처럼 처음

3

개

RU

는

64

개의 필터를 가지고 그다음

4

개

RU

는

128

개의 필터를 가지는

식입니다. 필터 개수가 이전

RU

와 동일할 경우는 ...