.h5文件转.pb文件(windows+Keras+Yolov3训练自己数据集)细节处理。
一、权重与网络结构转换的环境:
1.win10+cuda9.0
2.tensorflow-gpu==1.10;tensorflow==1.12
3.keras==2.2;python==3.5
二、Keras保存权重有两种方式:
#方式一:权重的参数和网络结构分开保存
1,保存网络结构到json文件
model_json=model.to_json()
with open("mosel.json") as file:
file.writ
原创
2020-06-17 13:49:27 ·
2520 阅读 ·
5 评论