runing some errors happens!
tomjeans opened this issue · 2 comments
when running the code,error happens!
Traceback (most recent call last):
File "demo.py", line 46, in
main()
File "demo.py", line 15, in main
det = Detector()
File "/home/nvidia/yolo-fatest-deepsort-main/Detector_pytorch.py", line 10, in init
self.init_model()
File "/home/nvidia/yolo-fatest-deepsort-main/Detector_pytorch.py", line 24, in init_model
load_darknet_weights(self.model, coco_weights)
File "/home/nvidia/yolo-fatest-deepsort-main/utils/models.py", line 394, in load_darknet_weights
conv.weight.data.copy_(torch.from_numpy(weights[ptr:ptr + nw]).view_as(conv.weight))
RuntimeError: shape '[96, 448, 1, 1]' is invalid for input of size 26694
Could you help me? Thanks.
Have you solved this problem? I've encountered the same question
It works after setting the correct cfg files !