search

phanxuanduc1996 / convert_yolo_weights

Summary of methods to convert models in Yolo-v4-v3-v2
73 stars 23 forks source link

tensorflow to onnx #3

Open qustions opened 4 years ago

qustions commented 4 years ago

Hi, @phanxuanduc1996 I am trying to convert ylov3.weights to TensorFlow and from Tensorflow to onnx but I did not find TensorFlow to onnx in this repo so I tried tf2onnx but there it's asking inputs and outputs python -m tf2onnx.convert --checkpoint tensorflow-model-meta-file-path --output model.onnx --inputs input0:0,input1:0 --outputs output0:0 when i ran it its giving error AssertionError: output0 is not in graph

phanxuanduc1996 commented 4 years ago

Hi @qustions , You can try to convert from .weights to .h5, then convert it to .onnx. Please check it in the source code.