search

david8862 / keras-YOLOv3-model-set

end-to-end YOLOv4/v3/v2 object detection pipeline, implemented on tf.keras with different technologies
MIT License
640 stars 222 forks source link

Can't convert model(.h5) to onnx format #247

Open Nand7773 opened 11 months ago

Nand7773 commented 11 months ago

I'm getting this error when running

export TF_KERAS=1

python keras_to_onnx.py

--keras_model_file="path/to/keras/model.h5"
--output_file="path/to/save/model.onnx"
--op_set=11

error is:- NameError: Exception encountered when calling layer "yolo_loss" (type Lambda). name 'K' is not defined