search

infocusp / tf_cnnvis

CNN visualization tool in TensorFlow
MIT License
780 stars 208 forks source link

Is it possible to only visualize one layer? #17

Closed zhmiao closed 7 years ago

zhmiao commented 7 years ago

by using the deconv_visualization() function, it generates deconv images for each layers by default. Is it possible to only generate images for one single layer? Thank you.

BhagyeshVikani commented 7 years ago

Hello @ranranking,

Yes. You can generate deconv images for only a specific layer by passing it in layers argument. Ex. If you want to generate deconv images for fc8 layer then pass layers=['fc8'].

Thank You

zhmiao commented 7 years ago

@BhagyeshVikani Thank you very much for the help! It works!

princeco commented 6 years ago

hello i want to know whether this tool support fc layer or not, i find it only support pooling,rely,and cone layer..