gmalivenko / onnx2keras

Convert ONNX model graph to Keras model format.
MIT License
195 stars 116 forks source link

Trying to convert from Yolov5 pt to onxx and then to h5 keras #164

Open SaulA01745146 opened 1 year ago

SaulA01745146 commented 1 year ago

I trained my model using yolov5 and with ultralitics/yolov5 github repository converted it to a onnx model. I want to convert it to a h5 keras model but the code throws an exception. `import onnx2keras from onnx2keras import onnx_to_keras import keras import onnx

onnx_model = onnx.load('best.onnx') k_model = onnx_to_keras(onnx_model,['images'],name_policy='renumerate')

keras.models.save_model(k_model,'kerasModel.h5',overwrite=True,include_optimizer=True)`

672d9b83-20aa-4b9b-9077-5f44fed28be7

Can someone tell me what can I do to fix this?