zuoshifan / NNhealpix

Neural networks on the Healpix sphere

NNhealpix

This is a package to apply neural networks and convolutional neural networks to spherical signal projected on the healpix grid.

Requirements

• Healpy
• Matplotlib
• NumPy
• Numba
• SciPy
• Tensorflow 2+

If you prefer to use PyTorch instead of Keras/TensorFlow, aasensio has implemented a port of the library available here: github.com/aasensio/sphericalCNN.

Install

The code is still under development. To install, use the following command:

[sudo] python setup.py develop [--user]

To automatically install the requirements, use the following command:

pip install -r requirements.txt

Testing

To run a suite of tests, you must have either nosetests or pytest. Just run nosetests or pytest within the NNhealpix folder.

License

The library is released under a MIT license. See the file LICENSE for more information.