search

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

something about the Summary #51

Open liuqinying opened 6 years ago

liuqinying commented 6 years ago

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?

naturomics commented 6 years ago

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.