如何从 .cfg 文件加载暗网 YOLOv3 模型并从 .weights 文件加载权重,并将带有权重的模型保存到 .h5 文件?
How to load darknet YOLOv3 model from .cfg file and weights from .weights file, and save the model with the weights to a .h5 file?
我已经从 darknet 下载了 YOLOv3 的 .weights 和 .cfg 文件(link:https://pjreddie.com/darknet/yolo/)我想创建一个模型并从这些文件分配权重,我想将具有指定权重的模型保存到 .h5 文件,以便我可以使用 keras.models.load_model() 将 .h5 模型加载到 Keras 中。
请帮忙。
您应该检查 this 存储库中给出的说明。这基本上是 YOLOv3(Tensorflow 后端)的 keras 实现。
- 从 YOLO 网站下载 YOLOv3 权重。
将 Darknet YOLO 模型转换为 Keras 模型。
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
由于您已经下载了权重和配置文件,您可以跳过第一步。从存储库下载 convert.py
脚本并简单地 运行 上面的命令。
注意:以上命令假定yolov3.cfg
、yolov3.weights
和model_data
(文件夹)与convert.py
。
我已经从 darknet 下载了 YOLOv3 的 .weights 和 .cfg 文件(link:https://pjreddie.com/darknet/yolo/)我想创建一个模型并从这些文件分配权重,我想将具有指定权重的模型保存到 .h5 文件,以便我可以使用 keras.models.load_model() 将 .h5 模型加载到 Keras 中。
请帮忙。
您应该检查 this 存储库中给出的说明。这基本上是 YOLOv3(Tensorflow 后端)的 keras 实现。
- 从 YOLO 网站下载 YOLOv3 权重。
将 Darknet YOLO 模型转换为 Keras 模型。
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
由于您已经下载了权重和配置文件,您可以跳过第一步。从存储库下载 convert.py
脚本并简单地 运行 上面的命令。
注意:以上命令假定yolov3.cfg
、yolov3.weights
和model_data
(文件夹)与convert.py
。