keras-YOLOv3-mobilenet icon indicating copy to clipboard operation
keras-YOLOv3-mobilenet copied to clipboard

convert .h5 to .pb

Open panguxiaoshen opened this issue 6 years ago • 3 comments

I have changed the structure of the model, and use "model.save()" to save the model. When I use "h5_model = load_model(weight_file_path, custom_objects={'yolo_loss': lambda y_true, y_pred: y_pred})", an error happended.

File "/home/ubuntu/lmy/program/keras-YOLOv3-mobilenet/yolo3/model.py", line 375, in yolo_loss grid, raw_pred, pred_xy, pred_wh = yolo_head(yolo_outputs[l], NameError: name 'yolo_head' is not defined

panguxiaoshen avatar Jun 13 '19 09:06 panguxiaoshen

@Adamdad 老铁求助!

panguxiaoshen avatar Jun 13 '19 10:06 panguxiaoshen

我在原版的keras-yolov3里看到了相关的解决办法 但是我看不懂 不知道老铁能不能参考解决下。。。 感觉有用的就是issues#48 #344

Rainweic avatar Jul 27 '19 02:07 Rainweic

我在原版的keras-yolov3里看到了相关的解决办法 但是我看不懂 不知道老铁能不能参考解决下。。。 感觉有用的就是issues#48 #344

同学请问你解决了吗

raozhongyu avatar Mar 16 '20 12:03 raozhongyu