CapsNet-Tensorflow
CapsNet-Tensorflow copied to clipboard
something about the Summary
In your code "capsNet.py",you add "self.decoded" to the "tf.summary.image" as "recon_img ",but self.X= input_image/255,and in your code " orgin = tf.reshape(self.X, shape=(cfg.batch_size, -1))
squared = tf.square(self.decoded - orgin)
self.reconstruction_err = tf.reduce_mean(squared)
" so self.decoded is not reconstructed image,you need to multiply it by 255,right?
tf.summary.image
can deal with this problem and display the images correctly. It's OK since we just want to display it on tensorboard
.