I am trying to convert yolo-tiny.weights to yolo-tiny.pth but I'm getting the following error can anyone please help
8858734
Traceback (most recent call last):
File "convert_darknet.py", line 243, in
load_weights(weightfile, yolov3, version)
File "convert_darknet.py", line 102, in load_weights
load_weights_yolov3(weightfile, yolov3)
File "convert_darknet.py", line 137, in load_weights_yolov3
ptr = copy_weights(bn, conv, ptr, weights)
File "convert_darknet.py", line 62, in copy_weights
conv_weights = conv_weights.view_as(conv.weight.data)
RuntimeError: shape '[512, 256, 3, 3]' is invalid for input of size 391694
I am trying to convert yolo-tiny.weights to yolo-tiny.pth but I'm getting the following error can anyone please help
8858734
Traceback (most recent call last):
File "convert_darknet.py", line 243, in
load_weights(weightfile, yolov3, version)
File "convert_darknet.py", line 102, in load_weights
load_weights_yolov3(weightfile, yolov3)
File "convert_darknet.py", line 137, in load_weights_yolov3
ptr = copy_weights(bn, conv, ptr, weights)
File "convert_darknet.py", line 62, in copy_weights
conv_weights = conv_weights.view_as(conv.weight.data)
RuntimeError: shape '[512, 256, 3, 3]' is invalid for input of size 391694