球面CNN解析源代码

s2cnn是一个来自《Spherical CNN》文章的代码实现包，默认情况下即可运行使用，并且功能上非常强大；不过其中最大的挑战在于理解其内部机制以及如何将其应用于实际场景中。完整代码仓库可参考：https://github.com/jonkhler/s2cnn

文件目录结构

首要的也是最优先的工具是setup.py这个脚本文件

此外包含了一个名为s2cnn的文件夹，在其中封装了论文中所述的各种卷积方法。这个文件夹中最重要的两个模块是位于软件目录下的s2_conv.py以及位于软件目录下的so3_conv.py这两个Python文件。它们分别实现了S2Convolution和SO3Convolution接口（实际上我们主要依赖的就是这两个接口）。这些接口还依赖于其他的一些Python代码模块，并且我们需要从S2Convolution这一部分开始进行深入学习（注意：所有的函数都可以通过Ctrl+B快速定位其定义位置）

S2Convolution

该方法接收的输入数据具有四维结构，在这一层中，各个维度分别对应于批次、通道、β值和α