더북(TheBook)

코드 5-32의 출력 결과인 (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), bias=False)에 대한 특성 맵을 확인해 보겠습니다.

코드 5-35 (0): Conv2d 특성 맵 확인

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

model(img)
activations = result.features

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

코드 5-36 특성 맵 확인

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()
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.