How to change the input size of the picture？

Question

How to change the input size of the picture？

mikaizhu opened this issue 4 years ago · 1 comments

mikaizhu commented 4 years ago

when my input image size is (3, 260, 260), then it went wrong....

RuntimeError:the size of tensor a(260) must match the size of tensor b(3) at non-singleton dimention 0

but when i resize the image (3, 256, 256), and it works.

so, how to change the input size of the picture？

Answer 1 · 2021-03-28T06:41:58.000Z

Hi,

You can change the image size but the image size should be 16^n, as we do three soft split with strides [4,2,2], and 4x2x2=16.
For you case, 260/16=16.25 is not an integer, 256/16=16 is an integer, so you can use 256 but not 260.