Как нарисовать многоугольник правильной формы из маски? - PullRequest
Новая
       96

Как нарисовать многоугольник правильной формы из маски?

0 голосов
/ 29 апреля 2020
  • У меня есть маска с различными помеченными областями, как показано на прикрепленном изображении.
  • После выполнения сегментации с активной функцией контура я хочу нарисовать многоугольник вокруг результирующих точек активных точек контура , но многоугольник считывает точки строка за строкой, поэтому рисунок многоугольника не является правильным, как видно на прилагаемом рисунке, многоугольник должен выглядеть больше как круглая область.

  • Как я могу это решить пожалуйста?

  • изображение

  • image
  • маска
  • Label
  • полигон
  • polygon
import numpy as np
from skimage.io import imread, imsave
import skimage.filters as filters
from skimage.segmentation import active_contour
from scipy.sparse import csr_matrix
import matplotlib.pyplot as plt
from skimage.draw import polygon
#---------------------------------
img  = imread('image.jpg')
mask = imread('label.png')

def activeContour(img, mask, alpha=0.015, beta=10, gamma=0.001, iterations=500, max_step=0.5):
    lb_list = np.unique(mask)
    Result = np.zeros_like(img)
    for lb in lb_list:
        if(lb != 0):
            s = np.where(mask == lb)
            r = s[0].astype(np.uint8)
            c = s[1].astype(np.uint8)
            init = np.array([r, c]).T
            #-------PreProcessing----------------------------
            img2 = filters.gaussian(img, 3)
            #------- Active Contour  ------------------------
            snake = active_contour(img2, init, alpha=alpha, beta=beta, gamma=gamma, coordinates='rc',
             max_px_move=max_step, max_iterations=iterations)
            #------------------------------------------------
            vals = lb*np.ones_like(snake[:, 0],dtype=np.uint8)
            label = csr_matrix((vals, (snake[:, 1].astype(np.uint8), snake[:, 0].astype(np.uint8))), shape=(img.shape[0], img.shape[1])).toarray()
            Result = Result + label
    return Result


res = activeContour(img, mask, alpha=0, beta= 10, gamma= 0.0000001)
labels = np.unique(res)
ctr = 25
for label in labels:
    if label != 0:
        img = np.zeros_like(img)
        s = np.where(res == label)
        r = s[0].astype(np.uint8)
        c = s[1].astype(np.uint8)
        rr, cc = polygon(r, c)
        img[rr, cc] = 2*ctr
        plt.imshow(img, cmap='gray')
        plt.title('%d'%label)
        plt.axis('off')
        plt.show()
        ctr+=1

1 Ответ

0 голосов
/ 30 апреля 2020

Спасибо за Хуану за помощь, теоретическая часть этого ответа здесь и реализация не работает. 

import numpy as np

def sortVertices(array):
    s = np.where(array != 0)
    yr = s[0].astype(np.uint8)
    xc = s[1].astype(np.uint8)
    center_xc = np.sum(xc)/xc.shape 
    center_yr = np.sum(yr)/yr.shape 
    theta = np.arctan2(yr-center_yr, xc-center_xc) * 180 / np.pi
    indices = np.argsort(theta)
    x = xc[indices]
    y = yr[indices]
    return x, y
...