CapsNet-Keras 项目使用教程

CapsNet-Keras 项目使用教程

CapsNet-KerasA Keras implementation of CapsNet in NIPS2017 paper "Dynamic Routing Between Capsules". Now test error ＝ 0.34%. 项目地址:https://gitcode.com/gh_mirrors/ca/CapsNet-Keras

1. 项目的目录结构及介绍

CapsNet-Keras/
├── capsulenet.py
├── capsulenet-multi-gpu.py
├── capsulelayers.py
├── utils.py
├── README.md
├── LICENSE
└── gitignore

• capsulenet.py: 主启动文件，用于训练和测试 CapsNet 模型。
• capsulenet-multi-gpu.py: 多 GPU 版本的启动文件。
• capsulelayers.py: 定义了 CapsNet 中的胶囊层。
• utils.py: 包含一些辅助函数和工具。
• README.md: 项目说明文档。
• LICENSE: 项目许可证。
• gitignore: Git 忽略文件配置。

2. 项目的启动文件介绍

capsulenet.py

这是项目的主启动文件，用于训练和测试 CapsNet 模型。以下是一些关键功能：

• 训练模型: 使用默认设置训练模型。

  python capsulenet.py
  

• 详细使用: 查看更多详细的使用选项。

  python capsulenet.py -h
  

capsulenet-multi-gpu.py

这是多 GPU 版本的启动文件，适用于需要更高计算资源的场景。

3. 项目的配置文件介绍

项目中没有显式的配置文件，但可以通过命令行参数在启动文件中进行配置。例如：

• 训练模型:

  python capsulenet.py --epochs 50 --batch_size 64
  

• 测试预训练模型:

  python capsulenet.py -t -w result/trained_model.h5
  

通过这些参数，可以灵活地调整训练和测试的设置。

CapsNet-KerasA Keras implementation of CapsNet in NIPS2017 paper "Dynamic Routing Between Capsules". Now test error ＝ 0.34%. 项目地址:https://gitcode.com/gh_mirrors/ca/CapsNet-Keras