PB模型转MNN模型以及MNN模型可视化

  |   0 评论   |   0 浏览

Tensorflow模型转MNN

参考MNN模型转换。

./MNNConvert -f TF --modelFile XXX.pb --MNNModel XXX.mnn --bizCode biz

MNN模型可视化

使用netron来进行模型可视化。

安装netron

pip install netron

使用

Python 3.7.10 (default, Feb 26 2021, 13:06:18) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import netron
>>> netron.start(None)
Serving at http://localhost:8080
('localhost', 8080)

会打开浏览器，在浏览器中上传模型，可得到结果，示例如下：