add EDAPs

minigrida/descriptors/edaps.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+# file edaps.py
+# author Florent Guiotte <florent.guiotte@irisa.fr>
+# version 0.0
+# date 07 juil. 2020
+"""Abstract
+
+doc.
+"""
+
+import numpy as np
+import sap
+from multiprocessing import Pool
+
+
+def _diff_attribute_profiles(*kwargs):
+    image = kwargs[0]
+    name = kwargs[3]
+    return sap.attribute_profiles(*kwargs).diff() \
+           + sap.Profiles([image[None]], 
+                          [{'tree': {'image_name': name}, 
+                            'attribute': 'altitude', 
+                            'profiles': [{'operation': 'copy'}]}
+                          ])
+
+
+def run(gt, rasters, coords, remove,
+        attributes, adjacency='4', filtering='direct', dtype=np.float32):
+    X = []
+    y = []
+    groups = []
+    Xn = None
+
+    for i, (gti, rastersi, coordsi) in enumerate(zip(gt, rasters, coords)):
+        # Compute EAP
+        attributes = [attributes] * len(rastersi) if isinstance(attributes, dict) else attributes
+        pool = Pool()
+        eap = pool.starmap(_diff_attribute_profiles, [
+                           (raster, attribute, adjacency, name, filtering)
+                           for (name, raster), attribute
+                           in zip(rastersi.items(), attributes)])
+        pool.close()
+        pool.join()
+        eap = sap.concatenate(eap)
+
+        Xn = [' '.join((a.description['tree']['image_name'],
+                a.description['attribute'],
+                *[sap.profiles._title(p)]))
+               for a in eap for p in a.description['profiles']] if not Xn else Xn
+
+        # Create vectors
+        X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1).astype(dtype)
+        y_raw = gti
+
+        # Remove unwanted label X, y
+        lbl = np.ones_like(y_raw, dtype=np.bool)
+        for l in remove if remove else []:
+            lbl &= y_raw != l
+
+        X += [X_raw[lbl]]
+        y += [y_raw[lbl]]
+        groups += [np.repeat(coordsi, lbl.sum())]
+
+    X = np.concatenate(X)
+    y = np.concatenate(y)
+    groups = np.concatenate(groups)
+
+    return X, y, groups, Xn