「TypeError：Expected Ptr <cv :: UMat> for argument '％s'」の修正方法

CUDAを使用してプログラムを高速化するコードを書いていますが、トリッキーなエラーが発生しました。私はそれについて何も知りません。私の環境はOpenCV 4.1.1、python 3.6です。これが私のコードです。

Imgを回転させる関数を定義します。

def rotate(img, angle):
    '''
    '''    
    if len(img.shape) == 3:
        (rows, cols, channels) = img.shape
        out_size = (cols, rows, channels)
    else:
        (rows, cols) = img.shape
        out_size = (cols, rows)

    if angle == 0:
        dst = img

    else:
        # img_gpu = cv2.cuda_GpuMat()
        img_gpu = cv2.cuda_GpuMat()
        out_gpu = cv2.cuda_GpuMat()
#         M_gpu = cv2.cuda_GpuMat()
#         out_size_gpu = cv2.cuda_GpuMat()
#         border_value_gpu = cv2.cuda_GpuMat()

        m = cv2.getRotationMatrix2D((cols/2, rows/2), angle, 1) 
        img_gpu.upload(img)
#         M_gpu.upload(M)
#         out_size_gpu.upload((12000, 6000))
#         border_value_gpu.upload((0, 0, 0))
        cols_gpu = cv2.cuda_GpuMat()
        rows_gpu = cv2.cuda_GpuMat()
        cols_gpu.upload(cols)
        rows_gpu.upload(rows)

        print(type(img))
        print(img.shape)
        (row,col) = img.shape
        print([img_gpu.size()[0],img_gpu.size()[1]])
#         M = np.float32([[1,0,100],[0,1,50]])
#         out=cv2.UMat(out_gpu,(284,284))
        out_gpu = cv2.cuda.warpAffine(img_gpu, m, (col,row))

        dst = out_gpu.download()

    return dst

それから私はそれを呼びます。

img = cv2.imread('../FengZhan/temp.png',0)
img_rotate = rotate(img, -10)

機能せず、次のエラーが発生します。

<ipython-input-48-41cd06952793> in rotate(img, angle)
     36 #         M = np.float32([[1,0,100],[0,1,50]])
     37 #         out=cv2.UMat(out_gpu,(284,284))
---> 38         out_gpu = cv2.cuda.warpAffine(img_gpu, m, (col,row))
     39 
     40         dst = out_gpu.download()

TypeError: Expected Ptr<cv::UMat> for argument '%s'

Img_gpuをcv2.UMat（img_gpu）に置き換えようとしましたが、まだ機能しません。誰かが私を助けますか？

python-3.xopencv

2019/08/06crazycao

私が OpenCV cuda bindings for pythonがcudawarping/src/warp.cpp＃239 。

ただし、このバグを修正しても、速度が満足のいくものではありません（CPUバージョンよりも遅い）。

幸運を。

2019/12/18miaodx