pip install tensorflow-gpu
pip install keras==2.0.9

git clone https://github.com/XifengGuo/CapsNet-Fashion-MNIST.git
cd CapsNet-Fashion-MNIST

$ python capsulenet.py

Stargazers

Forkers

tarrysingh baudm searchingmnist grseb9s rnnccl atifs 1kaiser zfxu harunpehlivan norbertbajko dawin2015 jzkay12 xiaoliang008 caprdzv shubhampachori12110095 specialistward anubhavagr27 ossdc yashvakil zwcdp censomin raajatk massoudkh29 raul1968 organicfool bigdatasciencegroup shulinyang rezacsedu iv461 liuwenhaha kims-github huqingli dansh-git yanxiankun suryapradeepm jane-one sharathchandrasuroj 4hck bhattacharjee jerrywly mohitajais peternara andysucao clarkgablewang ben745260

def test(model, data, args):
#    x_test, y_test = data
    y_pred, x_recon = model.predict([x_test, y_test], batch_size=100)
    print('Test acc:', np.sum(np.argmax(y_pred, 1) == np.argmax(y_test, 1))/y_test.shape[0])
    image = combine_images(np.concatenate([x_test[:50],x_recon[:50]]))
    image = image * 255
    Image.fromarray(image.astype(np.uint8)).save(args.save_dir + "/real_and_recon.png")
    print()
    print('Reconstructed images are saved to %s/real_and_recon.png' % args.save_dir)
    plt.imshow(plt.imread(args.save_dir + "/real_and_recon.png"))
    plt.show()