keras转onnx

最新推荐文章于 2024-08-16 08:28:10 发布

深度学习扛把子

最新推荐文章于 2024-08-16 08:28:10 发布

阅读量495

点赞数

分类专栏： python基础复习文章标签： keras 深度学习 python

本文链接：https://blog.csdn.net/qq_16792139/article/details/126015924

版权

python基础复习专栏收录该内容

27 篇文章 0 订阅

订阅专栏

from factory import NetworkFactory
import pkg_resources
from tensorflow.python.keras import backend as K
from tensorflow.python.keras.models import load_model
import onnx
import keras2onnx

weights_file = pkg_resources.resource_stream('mtcnn', 'mtcnn_weights.npy')
_pnet, _rnet, _onet = NetworkFactory().build_P_R_O_nets_from_file(weights_file)

onnx_model_name = 'model.onnx'

#training (bool, default False) - 在训练模式下导出模型。目前，ONNX导出的模型只是为了做推断，所以你通常不需要将其设置为True。
onnx_model = keras2onnx.convert_keras(_pnet, _pnet.name)
onnx.save_model(onnx_model, onnx_model_name)