Sizes of tensors must match except in dimension 2. Got 26 and 4

[khany27]

Hello,

I am getting this error, when I run video_demo.py.

Loading network..... Network successfully loaded Traceback (most recent call last): File "video_demo.py", line 111, in <module> model(get_test_input(inp_dim, CUDA), CUDA) File "C:\Users\Salem\anaconda3\envs\yo1\lib\site-packages\torch\nn\modules\module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "C:\Users\Salem\yolov3\darknet.py", line 341, in forward x = torch.cat((map1, map2), 1) RuntimeError: Sizes of tensors must match except in dimension 2. Got 26 and 4

[rttariverdi67]

Dear khany27 ,
did you solve this error? I faced with same !
for me it was the input image, like "scream.jpg" gives error!
bests.

[Wenqing-Zheng]

I met semilar problem, I solved it.
Date: 2020.12.2
how to fix:
in detect.py around line 224,
change:
prediction[:,0] += i*batch_size
into:

    if len(prediction)!=0:
        prediction[:,0] += i*batch_size

AND:

in util.py, around line 121:
change:
output = prediction.new(1, prediction.size(2) + 1)
into:
output = torch.tensor([])

in total, change two places, solved it.