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
関連記事
Facebookの顔認証技術『DeepFace』
UnityでTweenアニメーションを実装できる3種類の無料...
ZBrushトレーニング
Autodesk Mementoでゴジラを3次元復元する
ドラマ『ファーストクラス』のモーショングラフィックス
第1回 3D勉強会＠関東『SLAMチュートリアル大会』
株式会社ヘキサドライブの研究室ページ
このブログのデザインに飽きてきた
PyMC：Pythonのベイズ統計ライブラリ
fSpy：1枚の写真からカメラパラメーターを割り出すツール
Mechanizeで要認証Webサイトをスクレイピング
OpenCV 3.1から追加されたSfMモジュール
畳み込みニューラルネットワーク (CNN: Convolut...
マンガで分かる JavaScriptプログラミング講座
OpenFace：Deep Neural Networkによ...
JavaScriptとかWebGLとかCanvasとか
Google App EngineでWordPress
オンライン英会話ネイティブキャンプを始めてみた
Raspberry Pi 2を買いました
ManuelBastioniLAB：人体モデリングできるBl...
UnrealCLR：Unreal Engineで.NET C...
Pix2Pix：CGANによる画像変換
Alice Vision：オープンソースのPhotogram...
OpenCVで動画の手ぶれ補正
ブラウザ操作自動化ツール『Selenium』を試す
NeuralNetwork.NET：.NETで使えるTens...
RSSフィードを読込んで表示するWordpressプラグイン...
OANDAのfxTrade API
WordPress on Windows Azure
WordPressプラグインの作り方
画像認識による位置情報取得 - Semi-Direct Mo...
WordPress on Google App Engine...
Boost オープンソースライブラリ
Python for Unity：UnityEditorでP...
ベイズ推定とグラフィカルモデル
UnityのGameObjectの向きをScriptで制御す...
OpenMesh：オープンソースの３Dメッシュデータライブラ...
オープンソースのIT資産・ライセンス管理システム『Snipe...
GAN (Generative Adversarial Ne...
Kubric：機械学習用アノテーション付き動画生成パイプライ...
Dlib：C++の機械学習ライブラリ
YOLO (You Only Look Once)：ディープ...
コメント