yolov3将.weight模型转换为.pb时出错,有没有大佬,跪求解释帮助下

weights_num: 8678554
Traceback (most recent call last):
  File "from_darknet_weights_to_pb.py", line 13, in <module>
    load_ops = load_weights(tf.global_variables(), darknet_weights)
  File "D:\Downloads\tensorflow-yolov3-master\from_darknet_weights_to_ckpt.py", line 56, in load_weights
    (shape[3], shape[2], shape[0], shape[1]))
ValueError: cannot reshape array of size 211514 into shape (512,256,3,3)

从报错上面看,你的模型大小为211514,不够512*256*3*3的,估计你需要修改下输入参数和你的训练模型的参数输入一致才行