I have already finetuned an Inception V3 model on my dataset which was already pretrained on ImageNet. Now, I need to load the save pt model and run it on test patches and extract the 2048 feature vector from Inception V3. I am not sure what code snippet I should use for extracting feature vectors…

Yes it looks correct and also works fine: activation = {} def get_activation(name): def hook(model, input, output): activation[name] = output.detach() return hook model = models.inception_v3() model.avgpool.register_forward_hook(get_activation("avgpool")) x = torch.randn(2, 3, 299…

Extract the 2048 vector of a fine-tuned Inception V3 on test set

vision

Mona_Jalal (Mona Jalal) May 25, 2022, 3:44am 8

fixed it using module as suggested by you here