読者です 読者をやめる 読者になる 読者になる

walkingmask’s development log

IT系の情報などを適当に書いていきます

MENU

TensorFlowでCNNの各層のimage_summaryを作ってみた

講義の課題で TensorFlow Tutorial の "Deep MNIST for Experts" を触っていて,各レイヤのimage_summary(input image, kernel, output など)を作成したいと思って作ってみたのでlog.

出力画像

TensorBoardのIMAGESタグでimage_summaryの各種画像が見られるので,それらの画像.フィルタなどはすべて学習前の状態のものです.学習後どうなるのかはまた今度の楽しみに.

input 画像

f:id:walkingmask:20160816022821p:plain

第一層畳み込みフィルター

f:id:walkingmask:20160816022854p:plain

第一層畳み込み結果

f:id:walkingmask:20160816022914p:plain

第一層プーリング結果

f:id:walkingmask:20160816022942p:plain

第二層畳み込みフィルター

f:id:walkingmask:20160816023014p:plain

第二層畳み込み結果

f:id:walkingmask:20160816023023p:plain

第二層プーリング結果

f:id:walkingmask:20160816023036p:plain

第一層の畳み込みフィルタの画像ちっさいw
けどそれぞれの画像で何となく雰囲気は掴めるのではないかなと.学習後のものと比較すればもっと理解が深まりそう.

ソースコード

github.com

のsrc/にあります.ついでにimgに元画像も.

Visualizing CNNs with TensorBoard · Issue #908 · tensorflow/tensorflow · GitHub
Start on TensorBoard by RobRomijnders
conv neural network - Visualizing output of convolutional layer in tensorflow - Stack Overflow

このへんを参考にさせてもらって作ったけど,なかなか汚いコードでしかもまだ1つずつしか出力できないので,今後はもっと綺麗にかつ少しでもパフォーマンスを改善してマージしたいなと思います.

何かアドバイスがあればお待ちしてます!