Hi, I have been attempting to leverage the pre-trained ResNet model as a feature extractor. I have removed the final fc and pooling stages of the network and the output shape is (1, 2048, 7, 7).
This is not the type of features I want. What I want is a feature per pixel, i.e. an output of the following shape: (1, n_features, H, W) where H and W are the height and width of the input image.
Is there any intermediate layer which outputs this kind of shape in ResNet? Is there any other pre-trained model in pytorch which offers this?