Added ctis.inverters.NeuralNetworkInverter class, which trains a CNN to solve the inversion problem.

ctis/__init__.py

@@ -2,3 +2,8 @@
 A package for inverting imagery captured by a computed tomography imaging
 spectrograph.
 """
+from . import inverters
+
+__all__ = [
+    "inverters",
+]

ctis/inverters/__init__.py

@@ -0,0 +1,10 @@
+"""
+A collection of the different inversion algorithms defined in this package.
+"""
+from ._abc import AbstractInverter
+from ._cnn import NeuralNetworkInverter
+
+__all__ = [
+    "AbstractInverter",
+    "NeuralNetworkInverter",
+]

ctis/inverters/_abc.py

@@ -0,0 +1,42 @@
+import abc
+import dataclasses
+import named_arrays as na
+
+__all__ = [
+    "AbstractInverter",
+]
+
+
+@dataclasses.dataclass
+class AbstractInverter(
+    abc.ABC,
+):
+    """
+    An interface describing an algorithm which can invert CTIS observations
+    to yield a reconstruction of the observed scene.
+    """
+
+    def __call__(
+        self,
+        observation: na.FunctionArray[
+            na.SpectralPositionalVectorArray,
+            na.AbstractScalarArray,
+        ],
+        **kwargs,
+    ):
+        """
+        Invert the given image deprojections to reconstruct the original scene.
+
+        Parameters
+        ----------
+        observation
+            Observed images which have been projected backwards through the scene.
+            This defines the field of view and wavelength range of the
+            reconstructed scene.
+            We choose the deprojections instead of the images here since the
+            former defines the extent of the reconstructed scene without the
+            need for additional arguments.
+        kwargs
+            Additional keyword arguments which can be used by subclass
+            implementations.
+        """

ctis/inverters/_cnn/__init__.py

@@ -0,0 +1,9 @@
+"""
+An inversion algorithm based on training a convolutional neural network.
+"""
+
+from ._cnn import NeuralNetworkInverter
+
+__all__ = [
+    "NeuralNetworkInverter",
+]