我已经从darknet下载了YOLOv3的.weights和.cfg文件(链接:https://pjreddie.com/darknet/yolo/)。我想创建一个模型并分配这些文件中的权重,然后我想将带有分配权重的模型保存为.h5文件,以便我可以使用keras.models.load_model()将.h5模型加载到Keras中。
请帮助我。
回答:
你应该查看这个仓库中的说明。这基本上是YOLOv3的Keras实现(Tensorflow后端)。
- 从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
位于同一路径下。