tensorflow模型转ncnn模型

最新推荐文章于 2025-09-22 02:50:53 发布

转载最新推荐文章于 2025-09-22 02:50:53 发布 · 936 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jianfeifeng/p/11040173.html

文章标签：

#人工智能 #python

部署运行你感兴趣的模型镜像

　　ncnn本来是有tensorflow2ncnn的工具，但是在5月份时候被删除，原因是很多算子不支持，使用过程中很多bug，作者nihui直接将该功能删除。但是，tensorflow是目前最popular的深度学习框架，因此tensorflow转ncnn的需求还是必不可少的需求。下面提供一种将tensorflow转换为ncnn的一种解决方案。

感谢： https://github.com/Tencent/ncnn/issues/5

下面提供一个tf转ncnn的方法，在基于MobileNetV2修改的模型上测试通过，模型输出正确

tf2ncnn有很多op不支持，这里尝试tf2coreml2onnx2ncnn的方法进行解决

1、使用freeze_graph.py生成.pb模型；

2、使用tf-coreml将tf模型转换为coreml模型

例子：

import tfcoreml as tf_converter

tf_converter.convert(tf_model_path = './pfld.pb', mlmodel_path = 'my_model.mlmodel', output_feature_names = ['PFLDnet/fc8_1/BiasAdd:0'])

3、使用[WinMLTool]将coreml转换为onnx模型

https://docs.microsoft.com/zh-cn/windows/ai/windows-ml/convert-model-winmltools

from coremltools.models.utils import load_spec
# Load model file
model_coreml = load_spec('my_model.mlmodel') from winmltools import convert_coreml # Convert it! # The automatic code generator (mlgen) uses the name parameter to generate class names. model_onnx = convert_coreml(model_coreml, 7, name='ExampleModel') from winmltools.utils import save_model # Save the produced ONNX model in binary format save_model(model_onnx, 'example.onnx')