cv2 折线无法在 numpy 数组上绘制“参数 'img' 的预期 Ptr<cv::UMat>”

Question

我正在将图像和一系列点的 pytorch 张量表示形式转换为 numpy，以便我可以在点之间绘制线条并在 jupyter lab 中显示图像（使用 matplotlib）

如果我注释掉该cv2.polylines行，该代码将按预期工作，并向我显示图像。

    # convert img tensor to np

    img = img / 2 + 0.5

    img = img.detach().cpu().numpy().astype(np.float32)

    img = np.transpose(img, (1, 2, 0))

    print(type(img))

    # prints:

    # <class 'numpy.ndarray'>

    

    # convert label tensor to numpy ndarray

    pts = lbl.detach().cpu().numpy().reshape(-1, 1, 2)

    pts = np.rint(pts).astype(np.int32)

    print([pts])

    # prints:

    # [array([[[ 17, 153]],

    #        [[153, 154]],

    #        [[159, 692]],

    #        [[ 14, 691]]], dtype=int32)]

    

    # draw lines between the vertices in pts

    cv2.polylines(img, [pts], True, (0,255,255))

    # show the image with matplotlib.pyplot

    plt.imshow(img)

    plt.show()

但是折线给出了错误：

---> 36         cv2.polylines(img, [pts], True, (0,255,255))

     37         plt.imshow(img)

     38         plt.show()

TypeError: Expected Ptr<cv::UMat> for argument 'img'

如何在该图像上绘制线条？

Answer 1

解决方案是复制图像img&nbsp;=&nbsp;img.copy()看起来 Tensor.numpy() 实际上为您提供了底层数据结构的不可变视图，因此您可以显示它但不能修改它。要修改它，我必须创建自己的副本