TensorFlowでCNNの各層のimage_summaryを作ってみた
講義の課題で TensorFlow Tutorial の "Deep MNIST for Experts" を触っていて,各レイヤのimage_summary(input image, kernel, output など)を作成したいと思って作ってみたのでlog.
出力画像
TensorBoardのIMAGESタグでimage_summaryの各種画像が見られるので,それらの画像.フィルタなどはすべて学習前の状態のものです.学習後どうなるのかはまた今度の楽しみに.
input 画像
第一層畳み込みフィルター
第一層畳み込み結果
第一層プーリング結果
第二層畳み込みフィルター
第二層畳み込み結果
第二層プーリング結果
第一層の畳み込みフィルタの画像ちっさいw
けどそれぞれの画像で何となく雰囲気は掴めるのではないかなと.学習後のものと比較すればもっと理解が深まりそう.
ソースコード
の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つずつしか出力できないので,今後はもっと綺麗にかつ少しでもパフォーマンスを改善してマージしたいなと思います.
何かアドバイスがあればお待ちしてます!