From 5e9be93fa9e5749eed76260c2ed8b90984e49da0 Mon Sep 17 00:00:00 2001 From: Karamaz0V1 Date: Thu, 23 Aug 2018 15:14:46 +0200 Subject: [PATCH] Add triskele LFAPs in python wrapper --- python/triskele/Triskele.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/python/triskele/Triskele.py b/python/triskele/Triskele.py index 5c81555..8647f3b 100644 --- a/python/triskele/Triskele.py +++ b/python/triskele/Triskele.py @@ -37,8 +37,8 @@ class Triskele: self._write_infile(raster, dtype) - def filter(self, tree='max-tree', area=None, standard_deviation=None, moment_of_inertia=None): - self._setup(tree, area, standard_deviation, moment_of_inertia) + def filter(self, tree='max-tree', area=None, standard_deviation=None, moment_of_inertia=None, feature='weight'): + self._setup(tree, area, standard_deviation, moment_of_inertia, feature) self._run() return self._read_outfile() @@ -63,11 +63,12 @@ class Triskele: rasters = rasters.astype(dtype) write(self.infile, rasters) - def _setup(self, tree, area, standard_deviation, moment_of_inertia): + def _setup(self, tree, area, standard_deviation, moment_of_inertia, feature): self.process = [self.triskele_bin, '-i', '{}'.format(self.infile), '-o', '{}'.format(self.outfile), - '--{}'.format(tree)] + '--{}'.format(tree), + '--f-{}'.format(feature)] if area is not None: np.savetxt(self.areafile, area, fmt='%d')