├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── capsLayer.py ├── capsNet.py ├── config.py ├── eval.py ├── imgs ├── margin_loss.png ├── my_wechat_QR.png ├── reconstruction_loss.png ├── total_loss.png ├── training_loss.png └── wechat_group.png ├── results ├── test_000.png ├── test_005.png ├── test_010.png ├── test_015.png ├── test_020.png ├── test_025.png ├── test_030.png ├── test_035.png ├── test_040.png ├── test_045.png ├── test_050.png ├── test_055.png ├── test_060.png ├── test_065.png ├── test_070.png └── test_075.png ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | *__pycache__ 2 | data* 3 | .ropeproject 4 | logdir 5 | -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/README.md -------------------------------------------------------------------------------- /capsLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/capsLayer.py -------------------------------------------------------------------------------- /capsNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/capsNet.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/config.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/eval.py -------------------------------------------------------------------------------- /imgs/margin_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/margin_loss.png -------------------------------------------------------------------------------- /imgs/my_wechat_QR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/my_wechat_QR.png -------------------------------------------------------------------------------- /imgs/reconstruction_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/reconstruction_loss.png -------------------------------------------------------------------------------- /imgs/total_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/total_loss.png -------------------------------------------------------------------------------- /imgs/training_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/training_loss.png -------------------------------------------------------------------------------- /imgs/wechat_group.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/imgs/wechat_group.png -------------------------------------------------------------------------------- /results/test_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_000.png -------------------------------------------------------------------------------- /results/test_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_005.png -------------------------------------------------------------------------------- /results/test_010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_010.png -------------------------------------------------------------------------------- /results/test_015.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_015.png -------------------------------------------------------------------------------- /results/test_020.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_020.png -------------------------------------------------------------------------------- /results/test_025.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_025.png -------------------------------------------------------------------------------- /results/test_030.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_030.png -------------------------------------------------------------------------------- /results/test_035.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_035.png -------------------------------------------------------------------------------- /results/test_040.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_040.png -------------------------------------------------------------------------------- /results/test_045.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_045.png -------------------------------------------------------------------------------- /results/test_050.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_050.png -------------------------------------------------------------------------------- /results/test_055.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_055.png -------------------------------------------------------------------------------- /results/test_060.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_060.png -------------------------------------------------------------------------------- /results/test_065.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_065.png -------------------------------------------------------------------------------- /results/test_070.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_070.png -------------------------------------------------------------------------------- /results/test_075.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/results/test_075.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/debarko/CapsNet-Tensorflow/HEAD/utils.py --------------------------------------------------------------------------------