Simplified NL

[Shiro-LK]

Hi,

Thank you for your interesting papers and sharing the code.
I have a question about the paper, in particular the simplified NL module.

If I understand well, you are using the self attention in order to get some features which will permits you , then to weighted the different channel of your input images right ?

So if I want to code it from your code it will give :

context = self.spatial_pool(x) # dim context: NxCx1x1
output = conv2D(context) # conv1x1 with C input channels and C output channels , dim output NxCx1x1 

return x + output 

is that right ?

[xvjiarui]

Hi @Shiro-LK ,
Sorry for the late reply.
Yes, your implementation is correct.