search

NVIDIA / ContrastiveLosses4VRD

Implementation for the CVPR2019 paper "Graphical Contrastive Losses for Scene Graph Generation"
Other
199 stars 41 forks source link

How to get the visual result of the feature map in the paper? #9

Open waallf opened 4 years ago

waallf commented 4 years ago

Thanks for your code. I want to visual result of the feature map. Can you share the code with me?

jz462 commented 4 years ago

Hi @waallf,

I don't have the visualization script at hand, but it's not hard at all, you just need to obtain the output of the last convolution layer from the predicate branch, which is a feature map of shape 1 x C x H x W if you just input one image, then you take average along the channel dimension and the resultant H x W matrix is the visualized heat map that you want.

Ji

waallf commented 4 years ago

Thanks！