search

RobotEdh / Yolov-4

Yolo v4 using TensorFlow 2.x
110 stars 30 forks source link

Custom object parameter missing from loading yolov4.h5 model in yolov4-tf.ipynb #13

Open teakfi opened 3 years ago

teakfi commented 3 years ago

Error is fixed with using this:

yolo_model = load_model("models/yolo/yolov4.h5", custom_objects ={"Mish":Mish})

System: Windows 10 TensorFlow version: 2.4.1 Eager execution: True Keras version: 2.4.0

ps. I had to add tensorflow. in front of all keras imports and cuda version didn't work either