223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use scipy euclidean distance function instead of FMM (runs without skfmm 

module installed)
This commit is contained in:
Pratheek Shanthraj 2012-12-19 14:50:45 +00:00
parent 64d167fa90
commit a69aaa1315
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
processing/post/addEuclideanDistance.py
View File

@ -1,6 +1,6 @@
#!/usr/bin/env python #!/usr/bin/env python
import os,re,sys,math,numpy,skfmm,string,damask import os,re,sys,math,numpy,string,damask
from scipy import ndimage from scipy import ndimage
from optparse import OptionParser, Option from optparse import OptionParser, Option
@ -202,7 +202,7 @@ for file in files:
distance[i,:,:,:] = numpy.where(uniques > features[feature_id]['aliens'],0.0,1.0) distance[i,:,:,:] = numpy.where(uniques > features[feature_id]['aliens'],0.0,1.0)
for i in xrange(len(feature_list)): for i in xrange(len(feature_list)):
distance[i,:,:,:] = skfmm.distance(distance[i,:,:,:], dx=[unitlength]*3) distance[i,:,:,:] = ndimage.morphology.distance_transform_edt(distance[i,:,:,:])*[unitlength]*3
distance.shape = (len(feature_list),resolution.prod()) distance.shape = (len(feature_list),resolution.prod())
table.data_rewind() table.data_rewind()