Approximate and subdivide polygons¶

from __future__ import print_function

import numpy as np
import matplotlib.pyplot as plt

from skimage.draw import ellipse
from skimage.measure import find_contours, approximate_polygon, \
    subdivide_polygon


hand = np.array([[1.64516129, 1.16145833],
                 [1.64516129, 1.59375   ],
                 [1.35080645, 1.921875  ],
                 [1.375     , 2.18229167],
                 [1.68548387, 1.9375    ],
                 [1.60887097, 2.55208333],
                 [1.68548387, 2.69791667],
                 [1.76209677, 2.56770833],
                 [1.83064516, 1.97395833],
                 [1.89516129, 2.75      ],
                 [1.9516129 , 2.84895833],
                 [2.01209677, 2.76041667],
                 [1.99193548, 1.99479167],
                 [2.11290323, 2.63020833],
                 [2.2016129 , 2.734375  ],
                 [2.25403226, 2.60416667],
                 [2.14919355, 1.953125  ],
                 [2.30645161, 2.36979167],
                 [2.39112903, 2.36979167],
                 [2.41532258, 2.1875    ],
                 [2.1733871 , 1.703125  ],
                 [2.07782258, 1.16666667]])

# subdivide polygon using 2nd degree B-Splines
new_hand = hand.copy()
for _ in range(5):
    new_hand = subdivide_polygon(new_hand, degree=2, preserve_ends=True)

# approximate subdivided polygon with Douglas-Peucker algorithm
appr_hand = approximate_polygon(new_hand, tolerance=0.02)

print("Number of coordinates:", len(hand), len(new_hand), len(appr_hand))

fig, (ax1, ax2) = plt.subplots(ncols=2, figsize=(9, 4))

ax1.plot(hand[:, 0], hand[:, 1])
ax1.plot(new_hand[:, 0], new_hand[:, 1])
ax1.plot(appr_hand[:, 0], appr_hand[:, 1])


# create two ellipses in image
img = np.zeros((800, 800), 'int32')
rr, cc = ellipse(250, 250, 180, 230, img.shape)
img[rr, cc] = 1
rr, cc = ellipse(600, 600, 150, 90, img.shape)
img[rr, cc] = 1

plt.gray()
ax2.imshow(img)

# approximate / simplify coordinates of the two ellipses
for contour in find_contours(img, 0):
    coords = approximate_polygon(contour, tolerance=2.5)
    ax2.plot(coords[:, 1], coords[:, 0], '-r', linewidth=2)
    coords2 = approximate_polygon(contour, tolerance=39.5)
    ax2.plot(coords2[:, 1], coords2[:, 0], '-g', linewidth=2)
    print("Number of coordinates:", len(contour), len(coords), len(coords2))

ax2.axis((0, 800, 0, 800))

plt.show()

ZnJvbSBfX2Z1dHVyZV9fIGltcG9ydCBwcmludF9mdW5jdGlvbgoKaW1wb3J0IG51bXB5IGFzIG5wCmltcG9ydCBtYXRwbG90bGliLnB5cGxvdCBhcyBwbHQKCmZyb20gc2tpbWFnZS5kcmF3IGltcG9ydCBlbGxpcHNlCmZyb20gc2tpbWFnZS5tZWFzdXJlIGltcG9ydCBmaW5kX2NvbnRvdXJzLCBhcHByb3hpbWF0ZV9wb2x5Z29uLCBcCiAgICBzdWJkaXZpZGVfcG9seWdvbgoKCmhhbmQgPSBucC5hcnJheShbWzEuNjQ1MTYxMjksIDEuMTYxNDU4MzNdLAogICAgICAgICAgICAgICAgIFsxLjY0NTE2MTI5LCAxLjU5Mzc1ICAgXSwKICAgICAgICAgICAgICAgICBbMS4zNTA4MDY0NSwgMS45MjE4NzUgIF0sCiAgICAgICAgICAgICAgICAgWzEuMzc1ICAgICAsIDIuMTgyMjkxNjddLAogICAgICAgICAgICAgICAgIFsxLjY4NTQ4Mzg3LCAxLjkzNzUgICAgXSwKICAgICAgICAgICAgICAgICBbMS42MDg4NzA5NywgMi41NTIwODMzM10sCiAgICAgICAgICAgICAgICAgWzEuNjg1NDgzODcsIDIuNjk3OTE2NjddLAogICAgICAgICAgICAgICAgIFsxLjc2MjA5Njc3LCAyLjU2NzcwODMzXSwKICAgICAgICAgICAgICAgICBbMS44MzA2NDUxNiwgMS45NzM5NTgzM10sCiAgICAgICAgICAgICAgICAgWzEuODk1MTYxMjksIDIuNzUgICAgICBdLAogICAgICAgICAgICAgICAgIFsxLjk1MTYxMjkgLCAyLjg0ODk1ODMzXSwKICAgICAgICAgICAgICAgICBbMi4wMTIwOTY3NywgMi43NjA0MTY2N10sCiAgICAgICAgICAgICAgICAgWzEuOTkxOTM1NDgsIDEuOTk0NzkxNjddLAogICAgICAgICAgICAgICAgIFsyLjExMjkwMzIzLCAyLjYzMDIwODMzXSwKICAgICAgICAgICAgICAgICBbMi4yMDE2MTI5ICwgMi43MzQzNzUgIF0sCiAgICAgICAgICAgICAgICAgWzIuMjU0MDMyMjYsIDIuNjA0MTY2NjddLAogICAgICAgICAgICAgICAgIFsyLjE0OTE5MzU1LCAxLjk1MzEyNSAgXSwKICAgICAgICAgICAgICAgICBbMi4zMDY0NTE2MSwgMi4zNjk3OTE2N10sCiAgICAgICAgICAgICAgICAgWzIuMzkxMTI5MDMsIDIuMzY5NzkxNjddLAogICAgICAgICAgICAgICAgIFsyLjQxNTMyMjU4LCAyLjE4NzUgICAgXSwKICAgICAgICAgICAgICAgICBbMi4xNzMzODcxICwgMS43MDMxMjUgIF0sCiAgICAgICAgICAgICAgICAgWzIuMDc3ODIyNTgsIDEuMTY2NjY2NjddXSkKCiMgc3ViZGl2aWRlIHBvbHlnb24gdXNpbmcgMm5kIGRlZ3JlZSBCLVNwbGluZXMKbmV3X2hhbmQgPSBoYW5kLmNvcHkoKQpmb3IgXyBpbiByYW5nZSg1KToKICAgIG5ld19oYW5kID0gc3ViZGl2aWRlX3BvbHlnb24obmV3X2hhbmQsIGRlZ3JlZT0yLCBwcmVzZXJ2ZV9lbmRzPVRydWUpCgojIGFwcHJveGltYXRlIHN1YmRpdmlkZWQgcG9seWdvbiB3aXRoIERvdWdsYXMtUGV1Y2tlciBhbGdvcml0aG0KYXBwcl9oYW5kID0gYXBwcm94aW1hdGVfcG9seWdvbihuZXdfaGFuZCwgdG9sZXJhbmNlPTAuMDIpCgpwcmludCgiTnVtYmVyIG9mIGNvb3JkaW5hdGVzOiIsIGxlbihoYW5kKSwgbGVuKG5ld19oYW5kKSwgbGVuKGFwcHJfaGFuZCkpCgpmaWcsIChheDEsIGF4MikgPSBwbHQuc3VicGxvdHMobmNvbHM9MiwgZmlnc2l6ZT0oOSwgNCkpCgpheDEucGxvdChoYW5kWzosIDBdLCBoYW5kWzosIDFdKQpheDEucGxvdChuZXdfaGFuZFs6LCAwXSwgbmV3X2hhbmRbOiwgMV0pCmF4MS5wbG90KGFwcHJfaGFuZFs6LCAwXSwgYXBwcl9oYW5kWzosIDFdKQoKCiMgY3JlYXRlIHR3byBlbGxpcHNlcyBpbiBpbWFnZQppbWcgPSBucC56ZXJvcygoODAwLCA4MDApLCAnaW50MzInKQpyciwgY2MgPSBlbGxpcHNlKDI1MCwgMjUwLCAxODAsIDIzMCwgaW1nLnNoYXBlKQppbWdbcnIsIGNjXSA9IDEKcnIsIGNjID0gZWxsaXBzZSg2MDAsIDYwMCwgMTUwLCA5MCwgaW1nLnNoYXBlKQppbWdbcnIsIGNjXSA9IDEKCnBsdC5ncmF5KCkKYXgyLmltc2hvdyhpbWcpCgojIGFwcHJveGltYXRlIC8gc2ltcGxpZnkgY29vcmRpbmF0ZXMgb2YgdGhlIHR3byBlbGxpcHNlcwpmb3IgY29udG91ciBpbiBmaW5kX2NvbnRvdXJzKGltZywgMCk6CiAgICBjb29yZHMgPSBhcHByb3hpbWF0ZV9wb2x5Z29uKGNvbnRvdXIsIHRvbGVyYW5jZT0yLjUpCiAgICBheDIucGxvdChjb29yZHNbOiwgMV0sIGNvb3Jkc1s6LCAwXSwgJy1yJywgbGluZXdpZHRoPTIpCiAgICBjb29yZHMyID0gYXBwcm94aW1hdGVfcG9seWdvbihjb250b3VyLCB0b2xlcmFuY2U9MzkuNSkKICAgIGF4Mi5wbG90KGNvb3JkczJbOiwgMV0sIGNvb3JkczJbOiwgMF0sICctZycsIGxpbmV3aWR0aD0yKQogICAgcHJpbnQoIk51bWJlciBvZiBjb29yZGluYXRlczoiLCBsZW4oY29udG91ciksIGxlbihjb29yZHMpLCBsZW4oY29vcmRzMikpCgpheDIuYXhpcygoMCwgODAwLCAwLCA4MDApKQoKcGx0LnNob3coKQ==