Twitterで知ったブラウザ上でインタラクティブにCNN(Convolutional Neural Network)のネットワーク構造を3Dで可視化できるオープンソースのフレームワークTensorSpace.js。ネットワーク構造だけでなく、特徴マップも可視化できる。
Apache License 2.0だそうです。
TensorSpace.js
TensorSpaceは、TensorFlow.js, Three.js, Tween.jsで構築されたニューラルネットワークの3D視覚化フレームワークです。TensorSpaceは、ディープラーニングのレイヤー構築、訓練済みモデルの読み込み、ブラウザ上での3D可視化できるKerasライクなAPIを提供します。
TensorSpaceなら、モデルの構造とは何なのか、モデルはどのように訓練されているのか、モデルが中間情報に基づいてどのように結果を予測しているのかを直感的に学ぶことができます。
TensorSpaceでは、モデルの前処理を行うことでTensorFlow, Keras, TensorFlow.jsで事前に訓練したモデルの可視化をサポートしています。
例のごとく、ソースコードはGitHubにある↓
https://github.com/tensorspace-team/tensorspace
ドキュメントはこちら↓
https://tensorspace.org/html/docs/startIntro.html
ただ、自分で動作環境を構築しなくてもデモページで有名なニューラルネットワーク(LeNet, AlexNet, VGG-16, ResNet-50, MobileNetv1, YOLOv2-tiny, ACGAN)の可視化を試すことができる↓
https://tensorspace.org/html/playground/index.html
LeNet
AlexNet
YOLOv2-tiny
ResNet-50
ACGAN
オイラは仕組み・プロセスの可視化が大好物なので、こういう可視化ツールすごく好きです。
高橋 啓治郎さんのGANのUnity移植もそうだったけど、この可視化実装の敷居が下がっているのはTensorFlow.jsの存在がかなり効いているんだろうか。
https://qiita.com/keijiro/items/8874c5730feaf80b7943
関連記事
Raspberry Piでセンサーの常時稼働を検討する
RSSフィードを読込んで表示するWordpressプラグイン...
Perfumeのライブパフォーマンスのビジュアル
オープンソースの顔認識フレームワーク『OpenBR』
オープンソースのロボットアプリケーションフレームワーク『RO...
第25回コンピュータビジョン勉強会＠関東に行って来た
ROSの薄い本
WordPressプラグインの作り方
顔検出・認識のAPI・ライブラリ・ソフトウェアのリスト
Regard3D：オープンソースのStructure fro...
Raspberry Pi
PCA (主成分分析)
pythonもかじってみようかと
Photogrammetry (写真測量法)
3分の動画でプログラミングを学習できるサイト『ドットインスト...
OANDAのfxTrade API
R-CNN (Regions with CNN featur...
SSII 2014 デモンストレーションセッションのダイジェ...
Raspberry PiのGPIOを操作するPythonライ...
書籍『伝わる イラスト思考』読了
成果を待てない長学歴化の時代
3Dモデルを立体視で確認できるVRアプリを作っている
Polyscope：3Dデータ操作用GUIライブラリ
UnityプロジェクトをGitHubで管理する
UnityでARKit2.0
openMVG：複数視点画像から3次元形状を復元するライブラ...
Mayaのプラグイン開発
libigl：軽量なジオメトリ処理ライブラリ
動画で学ぶお絵かき講座『sensei』
Google App EngineでWordPress
UnityでTweenアニメーションを実装できる3種類の無料...
Webスクレイピングの勉強会に行ってきた
Kubric：機械学習用アノテーション付き動画生成パイプライ...
UnityでShaderの入力パラメータとして行列を渡す
法線マップを用意してCanvas上でShadingするサンプ...
Kinect for Windows v2の日本価格決定
CGレンダラ研究開発のためのフレームワーク『Lightmet...
Raspberry Pi 2のGPIOピン配置
組み込み向けのWindows OS 『Windows Emb...
iPadをハンディ3Dスキャナにするガジェット『iSense...
C++ 標準テンプレートライブラリ (STL)
Boost オープンソースライブラリ
コメント