ディープラーニングのネットワーク視覚化ツール(非常に軽量)

推奨ツール:なぜnetron
はニューラルネットワーク可視化ツールを使用する必要があるのですか?
他の人のモデルを複製するとき、モデルの入力名と出力名を知る必要がある場合がありますが、作成者が私たちに知らせない場合があるため、自分で確認する必要があります。このツールを使用すると、特定のネットワークの入力名と出力名を明確に確認できますネットワーク構造。テンソルボードよりも軽量で、さまざまなフレームワークをサポートしています。

netronでサポートされて
いるフレームワーク-対応するモデル:サポートされているフレームワークの
サポート:
ONNX(.onnx、.pb、.pbtxt)、Keras
(.h5、.keras)、
CoreML(.mlmodel)、
Caffe2(dict_net.pb predict_net .pbtxt)、
MXNet(.model、-symbol.json)
TensorFlow Lite(.tflite)。実験的な
サポート:
Caffe(.caffemodel、.prototxt)、
PyTorch(.pth)、
Torch(.t7)、
CNTK(.model 、.cntk)、
PaddlePaddle(モデル)、
ダークネット(.cfgの)、
scikit-はこちら(の.pkl)、
TensorFlow.js(model.json、.pb)
TensorFlow(.pb、.META、.pbtxt)。
上記のフレームワーク我々あなたはそれをテストすることができます:あなたはオンラインでモデルテストを見つけることができます

(1)インストール方法1、ソースコードのインストール
インストール方法:サポートlinux、windows、mac
表示:https : //github.com/lutzroeder/Netron
ツールをインストールします。pipでインストールすることをお勧めし
ます。ターミナル入力には非常に便利です:pip install中性子

(2)無料のインストール方法、オンラインバージョン
これはオンラインで視覚化されたバージョンであり、モデルがアップロードされている限り、構造を生成できます。https//lutzroeder.github.io/netron/

(1)インストール後のソースコードの利用方法
コード:

#coding=utf8
"""
@author xuwentao
dev 2020.1.17
可视化网络的工具代码
"""
# 导入可视化工具
import netron
model_path = "/home/xuwentao/AlphaPose/models/sppe/duc_se.pth"
netron.start(model_path)

実装:

(AlphaPose) xuwentao@xuwentao-OMEN-by-HP-Desktop-PC-880-p1xx:~/AlphaPose$ python view_model.py 
Serving '/home/xuwentao/AlphaPose/models/sppe/duc_se.pth' at http://localhost:8080

実行が完了すると、Webページバージョンの視覚化が自動的にポップアップし、モデルが正常に印刷されます。クレイジー開発

ここに画像の説明を挿入モデル構造の一部

129の元の記事を公開 43のような 100,000以上を訪問

おすすめ

転載: blog.csdn.net/nbxuwentao/article/details/104015308
おすすめ