Invalid argument0 in line 234

ayooshkathuria / pytorch-yolo-v3

A PyTorch implementation of the YOLO v3 object detection algorithm

3.3k stars 1.06k forks source link

Invalid argument0 in line 234 #143

Closed zoubaobao closed 4 years ago

zoubaobao commented 4 years ago

Traceback (most recent call last): File "detect.py", line 234, in output = torch.cat((output,prediction)) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 8 and 86 in dimension 1 at C:\w\1\s\tmp_conda_3.6_100715\conda\conda-bld\pytorch_1579082902617\work\aten\src\TH/generic/THTensor.cpp:612

zoubaobao commented 4 years ago

OK,there is something wrong in the last picture, I deleted it and the bug now is fixed. But, there is no output in the det document. I don't know why.

zoubaobao commented 4 years ago

I am on Windows and the problem has been solved. Please check other issues to find the answer. Thank you guys! You really help me a lot ;)