pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard

CAM using InceptionResnetV2

Open ghost opened this issue 6 years ago • 0 comments

Hi Cadene, Thanks a lot for providing these models. I am relatively new to PyTorch I am finding it difficult to generate Class Activation Maps for my trained model (InceptionResnetV2) on XRay images.

When i see some tutorial, I observe the following:

self.feature_extractor = model.module.densenet121.features
self.feature_extractor.eval()
self.classifier = model.module.densenet121.classifier

So here, type(feature_extractor) is torch.nn.modules.container.Sequential as the above code is for Densenet model taken from pytorch vision library. How do I generate features and classifier as the above code for InceptionResnetV2 from Pretrainedmodels?

ghost avatar Aug 20 '19 13:08 ghost