
Spherical CNNs的PyTorch实现-球面卷积网络-python
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目为Spherical CNNs在Pytorch框架下的实现,专注于开发和应用球面卷积神经网络,推动三维数据处理技术的进步。
Spherical CNNs:球面卷积网络的PyTorch实现
概述:
该库包含一个使用PyTorch实现的用于处理旋转对称性(如全向图像、地球信号等)的球形信号卷积神经网络(Spherical Convolutional Neural Networks, Spherical CNNs)。此实现基于文献[1]中的方法,适用于在球体和SO(3)空间中工作的等变CNN。
依赖项:
- PyTorch(>=0.4.0)
- cupy
- lie_learn
- pynvrtc
安装所有依赖的命令可以在新的conda环境中执行如下操作:
```
conda create --name cuda9 python=3.6
conda activate cuda9
# s2cnn deps
```
全部评论 (0)
还没有任何评论哟~


