opencv - 如何在Opencv中使用Pytorch Tensor对象而不转换为numpy数组？

Question

我正在尝试使用Python的Pytorch和Opencv开发文本检测应用程序。
我可以将Pytorch张量与Opencv一起使用，如下所示。

val = y[0,:,:,0].data.cpu().numpy()
cv2.threshold(val , 0.4, 1, 0)

但是需要很多时间。我需要通过使用张量对象来执行此操作。我怎样才能做到这一点？

Answer 1

鉴于您的阈值调用中的最后一个0表示 cv.THRESH_BINARY ，它将遵循以下功能:



当您的maxval设置为1时，您可以使用以下内容替换此阈值调用:

(y[0,:,:,0] > 0.4).float()

我要 float ，但是您可以根据需要更改ofc。甚至像这样:

(y[0,:,:,0] > 0.4).to(dtype=y.dtype)

因此它将保持相同的数据类型。