mengzmd / MS-CapsNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MS-CapsNet

A multi-scale capsule network

requirements

  • mxnet 1.1.0
  • pthon 3.6.0

Table of context

  1. mutliscaleCapsule.py
  2. msCapsuleLayer.py
  3. CapsuleLayer.py
  4. embedding.py

Results

Methods FashionMNIST(score) CIFAR10(score)
CapsNet 0.911 0.732
MS-CapsNet 0.922 0.751
MS-CapsNet+Drop 0.927 0.752

Citation

@ARTICLE{mscaps,
  author={C. {Xiang} and L. {Zhang} and Y. {Tang} and W. {Zou} and C. {Xu}},
  journal={IEEE Signal Processing Letters}, 
  title={MS-CapsNet: A Novel Multi-Scale Capsule Network}, 
  year={2018},
  volume={25},
  number={12},
  pages={1850-1854},
  doi={10.1109/LSP.2018.2873892}}

About


Languages

Language:Python 100.0%