在Python中将多边形覆盖在图像之上

Question

假设我在PIL中有一张图片

from PIL import Image
Image.open(path_to_my_image)

x点和y点的两个列表

x = ['10', '30', '70']
y = ['15', '45', '90']

有没有办法在此图像上用透明度覆盖多边形？

另外，PIL是否是一个不错的库？还是我应该使用另一个？（例如scikits.image，或使用pylab在图像中渲染）。

Answer 1

为此，您可以像这样使用Shapely和OpenCV：

import cv2
import numpy as np
from shapely.geometry import Polygon

x = [10, 30, 70]
y = [15, 45, 90]
alpha = 0.5 # that's your transparency factor
path = 'path_to_image.jpg'

polygon = Polygon([(x[0], y[0]), (x[1], y[1]), (x[2], y[2]), (x[0], y[2])])
int_coords = lambda x: np.array(x).round().astype(np.int32)
exterior = [int_coords(polygon.exterior.coords)]

image = cv2.imread(path)
overlay = image.copy()
cv2.fillPoly(overlay, exterior, color=(255, 255, 0))
cv2.addWeighted(overlay, alpha, image, 1 - alpha, 0, image)
cv2.imshow("Polygon", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

Answer 2

PIL是一个很好的工具：

import Image
import ImageDraw
img = Image.open(...).convert('RGBA')

x = ['10', '30', '70']
y = ['15', '45', '90']

# convert values to ints
x = map(int, x)
y = map(int, y)

img2 = img.copy()
draw = ImageDraw.Draw(img2)
draw.polygon(zip(x,y), fill = "wheat")

img3 = Image.blend(img, img2, 0.5)
img3.save('/tmp/out.png')

的呼叫签名为draw.polygon：

def polygon(self, xy, fill=None, outline=None):

因此，唯一的选择是fill和outline。我查看了源代码以找到此信息。

IPython告诉我：

In [38]: draw.polygon?
...
File:       /usr/lib/python2.7/dist-packages/PIL/ImageDraw.py

告诉我在哪里看。

要在的顶部绘制半透明的多边形img，请复制图像。一个副本，绘制没有Alpha的全彩色多边形。然后使用Image.blend将原始图像和副本与设置为的级别合并alpha。对于每个像素：

out = image1 * (1.0 - alpha) + image2 * alpha