为何PIL的ImageDraw.polygon不接受numpy array作为输入？

嘿，这个坑我之前踩过！核心问题在于PIL的polygon()方法并不直接支持numpy数组作为顶点输入——它对输入格式有明确要求，得是Python原生的序列类型才行。

为什么原来的元组版本能正常工作？ #

你最初定义的A = ((2,2), (2,302), ...)是嵌套元组，每个顶点都是标准的Python元组（包含普通int类型的x/y坐标），PIL可以直接遍历这个序列，逐个读取每个顶点的坐标值，自然能正确绘制多边形。

为什么numpy数组版本会出错？ #

当你传入numpy数组A = np.array([[2,2], ...], dtype="int32")时，PIL会把整个numpy数组当成一个“不可拆分的序列元素”来处理，而不是逐个解析里面的顶点。哪怕你用A.flatten()，得到的依然是numpy数组（元素是numpy的int32类型），PIL无法正确识别这些numpy原生类型的数值，最终导致绘图逻辑混乱，生成错误的掩码。

解决办法：转换成Python原生类型 #

只需要把numpy数组转换成PIL能识别的原生序列（列表/元组）即可，有两种常用方式：

方式1：转换成顶点序列（列表的列表/元组的元组）

import numpy as np
from PIL import Image, ImageDraw

A = np.array( [[ 2, 2], [ 2, 302], [302, 302], [302, 2]], dtype="int32" )
# 转换为Python列表的列表
A_python = A.tolist()

img = Image.new('L', (310, 310), 0)
ImageDraw.Draw(img).polygon(A_python, outline=1, fill=1)
mask = np.array(img)
print(mask)

方式2：展平后转换成原生数值列表

如果习惯用扁平的坐标序列，记得要把numpy数组转成Python原生列表：

A_flat = A.flatten().tolist()  # 关键是调用.tolist()转成原生列表
ImageDraw.Draw(img).polygon(A_flat, outline=1, fill=1)

补充说明 #

PIL的绘图API底层是基于Python原生类型实现的，对numpy数组的支持非常有限。所以在给PIL传递参数时，尽量转换成它熟悉的类型（普通int/float、列表、元组），就能避免这类兼容性问题。

内容的提问来源于stack exchange，提问作者matousc