keras 如何从.cfg文件加载darknet YOLOv3模型,从.weights文件加载权重,并将模型与权重保存到.h5文件中?

tquggr8v  于 2022-11-13  发布在  其他
关注(0)|答案(2)|浏览(239)

我已经从darknet下载了YOLOv3的.weights和.cfg文件(链接:https://pjreddie.com/darknet/yolo/)我想创建一个模型并从这些文件中分配权重,我想将分配了权重的模型保存到一个.h5文件中,这样我就可以通过使用keras.models.load_model()将.h5模型加载到Keras中。
请帮帮忙。

6g8kf2rb

6g8kf2rb1#

你应该查看this仓库中给出的说明。这基本上是YOLOv3(Tensorflow后端)的keras实现。

  • 从YOLO网站下载YOLOv3砝码。
  • 将暗网YOLO模型转换为Keras模型。

python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
由于您已经下载了权重和配置文件,因此可以跳过第一步,从存储库下载convert.py脚本,然后运行上面的命令。

:上述命令假定yolov3.cfgyolov3.weightsmodel_data(文件夹)与convert.py位于同一路径。

svmlkihl

svmlkihl2#

对于从这个得到错误的人尝试改变'convert.py'中的图层部分不确定是否是版本问题,但改变converter.py文件加载的方式'keras.layers'为我解决了所有错误

相关问题