taipingeric / yolo-v4-tf.keras

A simple tf.keras implementation of YOLO v4
MIT License
140 stars 79 forks source link

error while loading weights #21

Open Lasmar-Raya opened 2 years ago

Lasmar-Raya commented 2 years ago

i think this error occur when it comes to an output layer cannot reshape array of size 4243077 into shape (1024,512,3,3) ---> conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0])