将训练结果保存为.h5模型后无法用tf.keras.models.load_model加载

bubbliiiing / yolov4-tf2

这是一个yolo4-tf2（tensorflow2）的源码，可以用于训练自己的模型。

MIT License

314 stars 111 forks source link

Open Duton opened 2 years ago

Duton commented 2 years ago

修改save_weights_only = False将训练结果保存为模型后无法用tf.keras.models.load_model加载 2022-04-04 20-01-25 的屏幕截图

Duton commented 2 years ago

您好，我想请问怎样将训练结果保存为可以用于加载和部署的模型，我在加载保存的模型时碰到以下问题 2022-04-05 15-44-00 的屏幕截图 2022-04-05 16-01-54 的屏幕截图

bubbliiiing commented 2 years ago

loss不能这样保存吧，你可以预测的时候 save model

yxd117 commented 2 years ago

你好我用tf.keras.save_model 然后再load的时候也是这个问题看起来好像是这个get_anchors_and_decode有点问题

bubbliiiing commented 2 years ago

预测的时候，不是训练

yxd117 commented 2 years ago

对是预测的时候,大概发现问题了, load的时候需要import utils.bbox这个函数, 因为整个模型用了 custom objects. 可参考 https://www.tensorflow.org/guide/keras/save_and_serialize#custom_objects

bubbliiiing commented 2 years ago

0 0 哇hu