search

theAIGuysCode / yolov3_deepsort

Object tracking implemented with YOLOv3, Deep Sort and Tensorflow.
GNU General Public License v3.0
337 stars 186 forks source link

Error while loading custom yolov3 weights #65

Open jessiffmm opened 2 years ago

jessiffmm commented 2 years ago

I execute: python3 load_weights.py --weights ./weights/yolov3-voc_custom.weights --output ./weights/yolov3-custom.tf --num_classes 8

But I got the following error: bn_weights = bn_weights.reshape((4, filters))[[1, 0, 2, 3]] ValueError: cannot reshape array of size 28 into shape (4,32)

Is there a solution?