我已经从darknet下载了YOLOv3的.weights和.cfg文件(链接:https://pjreddie.com/darknet/yolo/)我想创建一个模型并从这些文件中分配权重,我想将分配了权重的模型保存到一个.h5文件中,这样我就可以通过使用keras.models.load_model()将.h5模型加载到Keras中。
请帮帮忙。
我已经从darknet下载了YOLOv3的.weights和.cfg文件(链接:https://pjreddie.com/darknet/yolo/)我想创建一个模型并从这些文件中分配权重,我想将分配了权重的模型保存到一个.h5文件中,这样我就可以通过使用keras.models.load_model()将.h5模型加载到Keras中。
请帮帮忙。
2条答案
按热度按时间6g8kf2rb1#
你应该查看this仓库中给出的说明。这基本上是YOLOv3(Tensorflow后端)的keras实现。
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
由于您已经下载了权重和配置文件,因此可以跳过第一步,从存储库下载
convert.py
脚本,然后运行上面的命令。注:上述命令假定
yolov3.cfg
、yolov3.weights
和model_data
(文件夹)与convert.py
位于同一路径。svmlkihl2#
对于从这个得到错误的人尝试改变'convert.py'中的图层部分不确定是否是版本问题,但改变converter.py文件加载的方式'keras.layers'为我解决了所有错误