How to visualize the feature map of each layer of the convolutional network

[Foolish (king)]

How do I replace the following code with my own network model?
########################################################
self.pretrained_model = models.vgg16(pretrained=True).features

[Eta_C]

1.Use register_forward_hook to get feature map

How to register_hook on Conv2d?

Hi @Umair_Javaid, Simply create a hook function and register it to the by calling register_forward_hook.

2. Use Captum or pytorch-cnn-visualizations to visualize the feature map

How to use heatmaps in predicting where the classifier is focusing on i need some help ,here is the code.how can i go about it?

Captum might give you a good toolset for model interpretability.

How to use heatmaps in predicting where the classifier is focusing on i need some help ,here is the code.how can i go about it?

Before Captum, I use pytorch-cnn-visualizations.