python - 在python中的cv::UMat::convertTo

Question

我想将UMat转换为其他类型。

我已经尝试过使用UMat方法convertTo，但出现以下错误:
AttributeError: 'cv2.UMat' object has no attribute 'convertTo'。

我也尝试使用numpy方法np.float32，该方法显然不适用于UMat:
TypeError: float() argument must be a string or a number, not 'cv2.UMat'
我将使用一个numpy数组，但我需要UMat格式的其他方法。

这是有问题的代码:

array255 = cv2.UMat(int(height), int(width), type=cv2.CV_8UC1, s=(255))

...

hsv_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

...

lhsv = cv2.split(hsv_frame)
s = cv2.subtract(array255, lhsv[1])
v = cv2.subtract(array255, lhsv[2])
#s = s.convertTo(cv2.CV_32F)
#v = v.convertTo(cv2.CV_32F)
s = np.float32(s)
v = np.float32(v)

Answer 1

UMat具有方法get()，返回numpy.array。调用它，而不是使用numpy类型转换方法。例如。

array255f = array255.get().astype('f')

比您可以将其转换回 UMat:

array255fum = cv2.UMat(array255f)

array255fum将为 cv2.CV_32F类型。