jiwoon-ahn / psa

Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentation, CVPR 2018
MIT License
385 stars 62 forks source link

Regarding the code #5

Closed laukun closed 6 years ago

laukun commented 6 years ago

Thank author for the code. I have two small questions:

  1. regarding this small block of code below to obtain cam: def forward_cam(self, x): x = super().forward(x) x = self.fc8(x) x = F.relu(x) x = torch.sqrt(x) return x what's x=torch.sqrt(x) for?
  2. regarding the pretrained caffe weghts, i.e. vgg16_20M.prototxt, on what dataset was it trained on? were you meant to use the deeplab_LargeFOV model of vgg16 version as the network to compute CAM, but why setting fc6_dilation=1 not 12 as in deeplab v1 paper?
jiwoon-ahn commented 6 years ago