How to extract feature map of every layers?

sacmehta / ESPNet

ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation

https://sacmehta.github.io/ESPNet/

MIT License

538 stars 111 forks source link

How to extract feature map of every layers? #50

Closed msson closed 5 years ago

msson commented 5 years ago

Hi,

I am trying to extract every (or specific) layer's feature map (actual rgb values). (but I can get each feature map's resolutions and filter's coefficients) I do not know how to figure it out, any suggestion?

Thanks.

sacmehta commented 5 years ago

One way is to create forward hooks. Other way is directly dump the outputs in forward function. Have you tried any of these?