skeleton code

host/filtering.py

 import cv2
 import numpy as np
+try:
     import trace_skeleton
+    use_swig = True
+except ImportError:
+    import trace_skeleton_old
+    use_swig = False
 import skimage.morphology
 import random
 
@@ -29,8 +34,11 @@ def stroke(img):
 
     im = skimage.morphology.skeletonize(img_binary).astype(np.uint8)
 
+    if use_swig:
+        polys = trace_skeleton.from_numpy(im)
+    else:
         rects = []
-    polys = trace_skeleton.traceSkeleton(im, 0, 0, im.shape[1], im.shape[0], 10, 999, rects)
+        polys = trace_skeleton_old.traceSkeleton(im, 0, 0, im.shape[1], im.shape[0], 10, 999, rects)
 
     for l in polys:
         # c = (200 * random.random(), 200 * random.random(), 200 * random.random())