더북(TheBook)

20번째 특성 맵의 결과 기존 고양이 이미지의 형태는 찾아볼 수 없게 되었습니다.

이번에는 40번째 계층에서 특성 맵 결과를 살펴봅시다.

코드 5-39 40번째 계층에 대한 특성 맵

result = LayerActivations(model.features, 40) ------ 40번째 Conv2d 특성 맵 확인

model(img)
activations = result.features

역시 특성 맵을 시각적으로 표현합니다.

코드 5-40 특성 맵 확인

fig, axes = plt.subplots(4, 4)
fig = plt.figure(figsize=(12,8))
fig.subplots_adjust(left=0, right=1, bottom=0, top=1, hspace=0.05, wspace=0.05)
for row in range(4):
    for column in range(4):
        axis = axes[row][column]
        axis.get_xaxis().set_ticks([])
        axis.get_yaxis().set_ticks([])
        axis.imshow(activations[0][row*10+column])
plt.show()
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.