SciPy Create 2D Polygon Mask - PullRequest
Новая
       44

SciPy Create 2D Polygon Mask

33 голосов
/ 07 сентября 2010

Мне нужно создать двумерный массив, который представляет двоичную маску многоугольника, используя стандартные пакеты Python.

  • вход: вершины многоугольника, размеры изображения
  • вывод: бинарныймаска многоугольника (двумерный массив)

(Большой контекст: я хочу получить преобразование расстояния этого многоугольника, используя scipy.ndimage.morphology.distance_transform_edt.)

Может кто-нибудь показатьмне как это сделать?

Ответы [ 5 ]

57 голосов
/ 17 сентября 2010

Ответ оказывается довольно простым: 

import numpy
from PIL import Image, ImageDraw

# polygon = [(x1,y1),(x2,y2),...] or [x1,y1,x2,y2,...]
# width = ?
# height = ?

img = Image.new('L', (width, height), 0)
ImageDraw.Draw(img).polygon(polygon, outline=1, fill=1)
mask = numpy.array(img)
24 голосов
/ 07 сентября 2010

В качестве чуть более прямой альтернативы ответу @ Anil, matplotlib имеет matplotlib.nxutils.points_inside_poly, который можно использовать для быстрой растеризации произвольного многоугольника.Например, 

import numpy as np
from matplotlib.nxutils import points_inside_poly

nx, ny = 10, 10
poly_verts = [(1,1), (5,1), (5,9),(3,2),(1,1)]

# Create vertex coordinates for each grid cell...
# (<0,0> is at the top left of the grid in this system)
x, y = np.meshgrid(np.arange(nx), np.arange(ny))
x, y = x.flatten(), y.flatten()

points = np.vstack((x,y)).T

grid = points_inside_poly(points, poly_verts)
grid = grid.reshape((ny,nx))

print grid

Что дает (массив логических значений): 

[[False False False False False False False False False False]
 [False  True  True  True  True False False False False False]
 [False False False  True  True False False False False False]
 [False False False False  True False False False False False]
 [False False False False  True False False False False False]
 [False False False False  True False False False False False]
 [False False False False False False False False False False]
 [False False False False False False False False False False]
 [False False False False False False False False False False]
 [False False False False False False False False False False]]

Вы должны быть в состоянии передать grid любой из функций scipy.ndimage.morphology довольно хорошо.

12 голосов
/ 21 апреля 2016

Обновление комментария Джо.API Matplotlib изменился с момента публикации комментария, и теперь вам нужно использовать метод, предоставляемый подмодулем matplotlib.path.

Рабочий код приведен ниже. 

import numpy as np
from matplotlib.path import Path

nx, ny = 10, 10
poly_verts = [(1,1), (5,1), (5,9),(3,2),(1,1)]

# Create vertex coordinates for each grid cell...
# (<0,0> is at the top left of the grid in this system)
x, y = np.meshgrid(np.arange(nx), np.arange(ny))
x, y = x.flatten(), y.flatten()

points = np.vstack((x,y)).T

path = Path(poly_verts)
grid = path.contains_points(points)
grid = grid.reshape((ny,nx))

print grid
4 голосов
/ 07 сентября 2010

Вы можете попробовать использовать библиотеку изображений Python, PIL. Сначала вы инициализируете холст. Затем вы создаете рисованный объект и начинаете создавать линии. Это предполагает, что многоугольник находится в R ^ 2 и что список вершин для ввода находится в правильном порядке.

Ввод = [(x1, y1), (x2, y2), ..., (xn, yn)], (ширина, высота) 

from PIL import Image, ImageDraw

img = Image.new('L', (width, height), 0)   # The Zero is to Specify Background Color
draw = ImageDraw.Draw(img)

for vertex in range(len(vertexlist)):
    startpoint = vertexlist[vertex]
    try: endpoint = vertexlist[vertex+1]
    except IndexError: endpoint = vertexlist[0] 
    # The exception means We have reached the end and need to complete the polygon
    draw.line((startpoint[0], startpoint[1], endpoint[0], endpoint[1]), fill=1)

# If you want the result as a single list
# You can make a two dimensional list or dictionary by iterating over the height and width variable
list(img.getdata())

# If you want the result as an actual Image
img.save('polgon.jpg', 'JPEG')

Это то, что вы искали, или вы спрашивали что-то другое?

3 голосов
/ 06 июля 2018

В качестве небольшой альтернативы @Yusuke N. ответ с использованием matplotlib.path, такой же эффективный, как и ответ с from PIL import Image, ImageDraw (не нужно устанавливать Pillow, не нужно учитывать integer или float полезно мне, ха?)

рабочий код ниже: 

import pylab as plt
import numpy as np
from matplotlib.path import Path

width, height=2000, 2000

polygon=[(0.1*width, 0.1*height), (0.15*width, 0.7*height), (0.8*width, 0.75*height), (0.72*width, 0.15*height)]
poly_path=Path(polygon)

x, y = np.mgrid[:height, :width]
coors=np.hstack((x.reshape(-1, 1), y.reshape(-1,1))) # coors.shape is (4000000,2)

mask = poly_path.contains_points(coors)
plt.imshow(mask.reshape(height, width))
plt.show()

И изображение результата ниже, где темная область равна False, яркая область равна True. enter image description here

...