florent/triskele
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add triskele LFAPs in python wrapper

Browse Source
This commit is contained in:
Florent Guiotte 2018-08-23 15:14:46 +02:00
parent 08dda96b06
commit 5e9be93fa9
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python/triskele/Triskele.py
View File

@ -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')