Как разделить изображение на '' n '' разных полигонов в Python - PullRequest
0 голосов
/ 03 ноября 2018

Я новичок в python и пытаюсь разделить изображение на 'n' другой полигон, используя python. Моя цель - преобразовать изображение в n случайных многоугольников. Я попробовал алгоритм Вороного, но он вроде грязный. Я был бы очень признателен за любую помощь. Любой другой метод сегментации и т. Д.

Мой предыдущий код:

import random
import matplotlib.pyplot as plt
import numpy as np
from scipy.spatial import Voronoi, voronoi_plot_2d

img = plt.imread("abc.jpg")
fig, ax = plt.subplots()
ax.imshow(img)

def points(radius,rangeX,rangeY,qty):
    deltas = set()
    for x in range(-radius, radius+1):
        for y in range(-radius, radius+1):
            if x*x + y*y <= radius*radius:
                deltas.add((x,y))

    randPoints = []
    excluded = set()
    i = 0
    while i<qty:
        x = random.randrange(*rangeX)
        y = random.randrange(*rangeY)
        if (x,y) in excluded: continue
        randPoints.append((x,y))
        i += 1
        excluded.update((x+dx, y+dy) for (dx,dy) in deltas)
    return randPoints



def plot1(randPoints,fig):
    points = np.array(randPoints)
    vor = Voronoi(points)
    print vor.vertices
    voronoi_plot_2d(vor,ax = fig.gca())
    #plt.savefig('abc.png')
    plt.show()

radius = 20
rangeX = (0, 960)
rangeY = (0, 480)
qty = 9

points = points(radius, rangeX, rangeY, qty)
plot1(points,fig)

Мой вклад: enter image description here

Мой вывод: enter image description here

Это для n = 9, я был бы признателен за любую помощь, которую я могу получить.

...