PSCapsNet:基于论文“通过Top-2分类评估卷积神经网络和胶囊网络用于图像分类的泛化能力”的PyTorch参数共享胶囊网...

PSCapsNet 基于的论文，一种参数共享胶囊网络的PyTorch实现。 要求 火炬 conda install pytorch torchvision -c pytorch PyTorchNet pip install git+https://github.com/pytorch/tnt.git@master 胶囊层 pip install git+https://github.com/leftthomas/CapsuleLayer.git@master OpenCV conda install opencv 数据集 我们已将数据集上传到（访问代码：tr4u）。 您无需自己下载数据集，代码将自动下载它们。 用法 火车模型 python -m visdom.server -logging_level WARNING & python main.py --data_type