Matrix-Capsules-EM-Tensorflow
Matrix-Capsules-EM-Tensorflow copied to clipboard
A Tensorflow implementation of CapsNet based on paper Matrix Capsules with EM Routing
I try to replace the batch_squash with batch_x, and the reconstruct loss going to very large. Why we can not training with original MNIST data?
eval.py
There is no file "eval.py" in the code files.
What happen if we remove `coord_add` function from code? why we need this function?
I completed the code for cifar10 and made it run. The model gets 0.37 accuracy which is very disheartening. Do you think there is a problem with the code or...
I run `./download.sh` under ./data and then the directory `smallNORB` and 4 files `smallnorb-xxxx-{testing|training}-{cat|dat}.mat` got available. But when I tried to train the network with the smallNORB dataset by sending...
Hi, thanks for sharing your implementation. I read through your high-level description on https://openreview.net/forum?id=HJWLfGWRb , and I have a question about your implementation of the EM routing for convolutional capsule...
Thanks for posting the code! I tried your code on smallNORM dataset with the parameter specified in the paper: A=64 B=8 C=D=16, routing iteration = 3, batch_size = 64 (set...
I want to realize the first normalization of smallNORB, but I could not get the value of mean and variance(std) in data/smallNORB.py 177 line: mean, variance = tf.nn.moments(image, [0, 1,...
Could you please explain why loss values after 1200 iterations are growing up? 2018-12-28 07:17:58,417 [19476] INFO main: 412 iteration finishs in 3.548040 second loss=0.288399 2018-12-28 08:17:33,187 [19476] INFO main:...
I want to train this program with my own dataset, but I have difficulties to add it to the get_coord_add function. Could someone please help me ?