search

OliverRensu / RTNet

24 stars 3 forks source link

The H and W of the input img of this function SelfScaling are 1 respectively? #1

Open longmalongma opened 3 years ago

longmalongma commented 3 years ago

Thanks for your good work. The H and W of the input img of this function SelfScaling are 1 respectively? Otherwise, mul cannot be executed.

image

OliverRensu commented 3 years ago

Hi, the average pooling applies to the image which makes the H and W of the image become 1, therefore, image can be any size.

longmalongma commented 3 years ago

Hi, the average pooling applies to the image which makes the H and W of the image become 1, therefore, image can be any size.

@OliverRensu The mul requires the two matrices to be the same. After pooling, the H and W become 1. If the image can be any size, the dimensions of the two matrices are different. How to perform mul?

OliverRensu commented 3 years ago

The size of the image is H and W, and the size of f is 1 and 1. The broadcast mechanism in torch will automatically extend the size of f to H and W. For more information about broadcast you can visit https://pytorch.org/docs/stable/notes/broadcasting.html