keras-YOLOv3-mobilenet
keras-YOLOv3-mobilenet copied to clipboard
convert .h5 to .pb
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
@Adamdad 老铁求助!
我在原版的keras-yolov3里看到了相关的解决办法 但是我看不懂 不知道老铁能不能参考解决下。。。 感觉有用的就是issues#48 #344