-
Notifications
You must be signed in to change notification settings - Fork 0
/
SkeletonProg.py
29 lines (23 loc) · 797 Bytes
/
SkeletonProg.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import sys
import imageio
import Skeleton
def main(ifile_name, ofile_name1, bin_fn="bfile.bin", svg_file=None):
im = imageio.imread(ifile_name,as_gray=True)
skeleton = Skeleton.Skeleton(im)
skeleton.segments.addInitialStartPt()
skeleton.euclidMstOrder()
skeleton.segments.concatSegments()
#skeleton.segments.simplify()
skeleton.segments.segment2grad(interior=True)
skeleton.segments.renderGrad()
im = skeleton.segments.grad
imageio.imwrite(ofile_name1, im)
skeleton.segments.scaleBin()
skeleton.segments.binWrite(bin_fn)
if __name__ == "__main__":
if len(sys.argv) == 4:
main(sys.argv[1], sys.argv[2], sys.argv[3])
elif len(sys.argv) == 3:
main(sys.argv[1], sys.argv[2])
else:
print("Error: unknown usage")