Add feature profiles

minigrida/descriptors/fps.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+# file fps.py
+# author Florent Guiotte <florent.guiotte@irisa.fr>
+# version 0.0
+# date 31 mai 2020
+"""Abstract
+
+doc.
+"""
+
+import numpy as np
+import sap
+from joblib import Memory
+
+memory = Memory(location='cache/', verbose=0)
+
+
+@memory.cache
+def _feature_profiles(*kwargs):
+    return sap.feature_profiles(*kwargs)
+
+
+def run(gt, rasters, coords, remove, attributes, adjacency='4',
+        feature='same', filtering='direct'):
+    X = []
+    y = []
+    groups = []
+    Xn = None
+
+    for i, (gti, rastersi, coordsi) in enumerate(zip(gt, rasters, coords)):
+        # Compute EAP
+        eap = []
+        for name, raster in rastersi.items():
+            eap += [_feature_profiles(image=raster,
+                                      attribute=attributes,
+                                      adjacency=adjacency,
+                                      image_name=name,
+                                      out_feature=feature,
+                                      filtering_rule=filtering)]
+        eap = sap.concatenate(eap)
+
+        Xn = [' '.join((a['tree']['image_name'],
+                        a['attribute'],
+                        'feature',
+                        a['out feature'],
+                        *[str(v) for v in p.values()]))
+                       for a in eap.description for p in a['profiles']] if not Xn else Xn
+
+        # Create vectors
+        X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1)
+        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