我已经从暗网下载了 YOLOv3 的 .weights 和.cfg文件(链接： https://pjreddie.com/darknet/yolo/( 我想创建一个模型并从这些文件分配权重，我想将带有分配权重的模型保存到 .h5 文件中，以便我可以使用 keras.models.load_model(( 将 .h5 模型加载到 Keras 中。
请帮忙。
您应该检查此存储库中给出的说明。这基本上是YOLOv3(Tensorflow后端(的keras实现。
- 从 YOLO 网站下载 YOLOv3 砝码。
-
将暗网 YOLO 模型转换为 Keras 模型。
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
由于您已经下载了权重和配置文件，因此可以跳过第一步。从存储库下载convert.py脚本，然后只需运行上述命令即可。
注意：上述命令假定yolov3.cfg、yolov3.weights和model_data(文件夹(与convert.py位于同一路径中。
对于从中得到错误的人，请尝试更改"convert.py"中的图层部分不确定这是否是版本问题，但更改文件加载"keras.layers"的方式 converter.py 为我解决了所有错误