ENH: more efficient binary operation in skeletonize

2026-07-12 14:00:47 +08:00 · 2011-10-22 12:10:13 +02:00
parent c48aa63323
commit c45db9f4cc
1 changed files with 1 additions and 1 deletions
@@ -127,7 +127,7 @@ def skeletonize(image):

        # pass 2 - remove the 2's and 3's
        neighbours = correlate(skeleton, mask, mode='constant')
-        neighbours[skeleton == 0] = 0
+        neighbours *= skeleton
        codes = np.take(lut, neighbours)
        code_mask = (codes == 2)
        if np.any(code_mask):