
PSCapsNet: 一种基于PyTorch的参数共享胶囊网络,源自通过Top-2分类评估卷积神经网络和胶囊网络在图像分类中的泛化性能论文的研究成果。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
PSCapsNet是基于PyTorch开发的一种参数共享胶囊网络,该模型源于研究论文《通过Top-2分类评估卷积神经网络和胶囊网络在图像分类中的泛化性能》,旨在提高图像分类任务中的泛化能力。
PSCapsNet 是一种参数共享胶囊网络的 PyTorch 实现。安装火炬环境可以使用以下命令:`conda install pytorch torchvision -c pytorch` 或者 `pip install git+https://github.com/pytorch/tnt.git@master` 安装胶囊层可使用该命令:`pip install git+https://github.com/leftthomas/CapsuleLayer.git@master`。OpenCV 的安装可以使用 `conda install opencv` 命令。
数据集已上传,您无需自行下载,代码会自动处理这一部分的工作。为了训练模型,请按照以下步骤操作:启动 Visdom 服务器:`python -m visdom.server -logging_level WARNING &` 然后运行主程序: `python main.py --data_type`
全部评论 (0)
还没有任何评论哟~


