How to convert yolov4-custom from pretrain ( yolov4.conv.137) to .pb ?

theAIGuysCode / yolov4-deepsort

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.

GNU General Public License v3.0

1.32k stars 751 forks source link

How to convert yolov4-custom from pretrain ( yolov4.conv.137) to .pb ? #10

Open mops1112 opened 4 years ago

mops1112 commented 4 years ago

i already get final weight (custom my data) from pretrain yolo4.con.137. but i cant convert to .pb. it show error on save_model.py conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 4554552 into shape (1024,512,3,3)

George-Ogden commented 3 years ago

Edit the config.py files so that you have the correct classes file and then try to reload it.

ShahrozTanveer commented 3 years ago

hey @mops1112 , did you managed to solve this problem?

MartinBaezaCo commented 2 years ago

Hey there, I was going trough the same issue, are you using the tiny dataset? I had to manually chage the code on "save_model.py" line 10 to true, I'm really new to coding so I'm not sure if this could be set up somewhere else. But After changing from false to true, everything worked perfectly