opencv2.fillPoly的python接口想要什么作为输入?

时间:2012-06-29 23:42:56

标签: python opencv drawing

我正在尝试使用python接口绘制一个多边形来打开opencv,cv2。我创建了一个空图像,只是一个640x480的numpy数组。我有一个我想要在图像上绘制的多边形(四点四边形)列表,但是,我似乎无法使合成器正确指示cv2四边形应该在哪里,并且我不断收到此错误:

OpenCV Error: Assertion failed (points.checkVector(2, CV_32S) >= 0) in fillConvexPoly, file .../OpenCV-2.4.0/modules/core/src/drawing.cpp, line 2017

我的代码主要包含以下内容:

binary_image = np.zeros(image.shape,dtype='int8')
for rect in expected:
    print(np.array(rect['boundary']))
    cv2.fillConvexPoly(binary_image, np.array(rect['boundary']), 255)
fig = pyplot.figure(figsize=(16, 14))
ax = fig.add_subplot(111)
ax.imshow(binary_image)
pyplot.show()

其中我的预期rects列表中的'boundary'包含(x,y)点列表的值。代码打印:

[[ 91 233]
 [419 227]
 [410 324]
 [ 94 349]]

我认为这是多边形的点列表,但显然该列表具有无效的points.checkvector,无论是什么。谷歌搜索该错误没有任何用处。

4 个答案:

答案 0 :(得分:16)

import numpy as np
import cv2
import matplotlib.pyplot as plt

a3 = np.array( [[[10,10],[100,10],[100,100],[10,100]]], dtype=np.int32 )
im = np.zeros([240,320],dtype=np.uint8)
cv2.fillPoly( im, a3, 255 )

plt.imshow(im)
plt.show()

result display

答案 1 :(得分:8)

AssertionError告诉您OpenCV需要一个带符号的32位整数。多边形点阵列应具有该特定数据类型(例如points = numpy.array(A,dtype='int32'))。您也可以将其投射到函数调用(即my_array.astype('int32'))或朋友将其放置一次......

” 改变

cv2.fillConvexPoly(binary_image, np.array(rect['boundary']), 255)

cv2.fillConvexPoly(binary_image, np.array(rect['boundary'], 'int32'), 255)

答案 2 :(得分:6)

我在opencv 2.4.2和python 2.7中尝试过。 来自c ++界面

void fillPoly(Mat& img, 
              const Point** pts, 
              const int* npts, 
              int ncontours, 
              const Scalar& color, 
              int lineType=8, 
              int shift=0, 
              Point offset=Point() 
             )

我们知道 pts 是点数组的数组,所以你应该像这样改变

cv2.fillConvexPoly(binary_image, np.array([rect['boundary']], 'int32'), 255)

将[]添加到 rect ['boundary']

答案 3 :(得分:0)

这是一个带有注释点的示例。您可以按顺时针或逆时针顺序指定多边形顶点,只要它们遵循相同的方向即可。

import numpy as np
import matplotlib.pyplot as plt
import cv2

A = np.array([125,50])
B = np.array([50,50])
C = np.array([50,175])
D = np.array([150,150])

pts = np.array( [ A, B, C, D ] ) # counter-clockwise
#pts = np.array( [ A, D, C, B ] ) # clockwise

print(pts.shape) # (4, 2)

image = np.zeros((200,200), dtype=np.uint8)
image = np.dstack((image, image, image)) # Create three channels.

cv2.fillPoly(image, pts=[pts], color =(0,255,0))

for pt in pts:
    x = pt[0]
    y = pt[1]
    _ = plt.annotate(s='%d, %d' % (x, y), xy=(x, y), color='red', fontsize=20)

print(image.shape) # (200, 200, 3)

plt.imshow(image)
plt.grid(True)
plt.show()

enter image description here

如果您的顶点重新排列,使它们不遵循相同的方向,会发生什么?

pts = np.array( [ A, B, D, C ] )

你明白了:

enter image description here