arrays - cv.fillPoly生成零数组，不读取输入

Question

我有
blank = np.zeros(shape = im.shape, dtype = np.float32)
产生

array([[0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       ...,
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.]], dtype=float32)

我的[label2poly[label]]是dtype('int32')

[array([[ 716,    1],
        [ 710,  281],
        [ 727,  322],
        [ 756,  369],
        [ 793,  399],
        [ 863,  406],
        [ 952,  416],
        [ 978,  412],
        [ 416,    1]])]

当我尝试cv2.fillPoly(blank, [label2poly[label]], 255)时，输出

array([[0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       ...,
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.]], dtype=float32)`

应该是这样的地方

array([[  0.,   0.,   0., ...,   0.,   0.,   0.],
       [  0.,   0., 255., ...,   0.,   0.,   0.],
       [  0.,   0., 255., ...,   0.,   0.,   0.],
       ...,
       [  0., 255., 255., ..., 255., 255., 255.],
       [  0., 255., 255., ...,   0.,   0.,   0.],
       [  0.,   0.,   0., ...,   0.,   0.,   0.]], dtype=float32)

我正在尝试创建多个蒙版。将不胜感激。

Answer 1

blank的初始化:

blank = np.zeros(shape=[5, 5], dtype=np.float32)
print(blank)

输出:

[[0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]]

初始化label，然后调用fillPoly()，该调用使用mask作为在特定位置用 255 填充blank的掩码:

label = np.array([[[1, 1], [1, 2], [1, 3], [2, 1], [2, 2], [2, 3], [3, 1], [3, 2], [3, 3]]], dtype=np.int32)

cv2.fillPoly(blank, [label], 255)
print(blank)

输出:

[[  0.   0.   0.   0.   0.]
 [  0. 255. 255. 255.   0.]
 [  0. 255. 255. 255.   0.]
 [  0. 255. 255. 255.   0.]
 [  0.   0.   0.   0.   0.]]