Made SpatialTransformer layer serializable + ignore models/ dir

.gitignore

@@ -7,3 +7,4 @@ venv/
 *.egg-info/
 *.h5
 *.log
+models/

ddmr/layers/SpatialTransformer.py

@@ -3,10 +3,10 @@ import tensorflow.keras.backend as K
 import tensorflow as tf
 import neurite as ne
 
-import h5py
 from ddmr.utils.constants import IMG_SHAPE, DISP_MAP_SHAPE
 
 
+@tf.keras.utils.register_keras_serializable()
 class SpatialTransformer(kl.Layer):
     """
     Adapted SpatialTransformer layer taken from VoxelMorph v0.1