【发布时间】:2021-01-04 14:57:53
【问题描述】:
我有 yolov4 小权重和 cfg 文件,有什么方法可以将它们转换为 h5 文件吗?
【问题讨论】:
标签: deep-learning computer-vision yolo
我有 yolov4 小权重和 cfg 文件,有什么方法可以将它们转换为 h5 文件吗?
【问题讨论】:
标签: deep-learning computer-vision yolo
要将权重从一种格式转换为另一种格式,您首先需要在模型中加载权重。
将权重转换为浮点值。这意味着您需要读取模型每一层的权重
然后使用特定库(在您的情况下为 Keras)将权重保存为您想要的格式。
您可以参考以下链接了解如何将权重转换为 .h5
http://dsbyprateekg.blogspot.com/2020/06/how-to-convert-your-yolov4-weights-to.html?m=1
【讨论】: