더북(TheBook)

다음 그림은 첫 번째 Conv2d 계층에서 특성 맵에 대한 출력 결과입니다.

▲ 그림 5-47 첫 번째 계층에서 특성 맵

입력층과 가까운 계층으로 입력 이미지의 형태가 많이 유지되고 있습니다. 학습이 진행될수록 이미지의 변화를 확인할 수 있도록 20번째 계층과 40번째 계층에 대한 특성 맵도 함께 살펴보겠습니다.

먼저 20번째 계층에 대한 특성 맵을 살펴봅시다.

코드 5-37 20번째 계층에 대한 특성 맵

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

model(img)
activations = result.features
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.