aliyun/NeWCRFs

RuntimeError: shape '[1, 20, 7, 35, 7, 256]' is invalid for input of size 8843520

kingoo123 opened this issue · 8 comments

Hi,I'm trying to test on my own test set, but the following issues arise and I would like to ask you how to fix them
The environment used is as follows:
python 3.8
cuda 10.1
pytorch 1.8.1
mmcv 1.4.8
image

Maybe related to the image size of your own data

So,how to test own data?

So,how to test own data?
I also want to know it.

It can be directly used on your own data. But you need to fix the bug.

The reason for the error is that your image H,W is not divisible by 32 during the network downsampling process. you can refer to the yolo-v5--def letterbox to change the H,W to be divisible by 32.

6leet commented

Hi, I'm facing this problem currently, is there any general solution yet? Thanks!

is there any solution to this problem