Как запустить CV2 с CUDA? Для калибровки камеры. (Python) - PullRequest
Новая
       71

Как запустить CV2 с CUDA? Для калибровки камеры. (Python)

0 голосов
/ 29 апреля 2020

Я пытаюсь откалибровать свою камеру. Этот раздел:

cv2.calibrateCamera

занимает слишком много времени для вычисления матрицы и других параметров, поэтому мне интересно, как я могу запустить ее на CUDA ?!

Я использую Nvidia Gforce 1050TI. Это основная и «тяжелая» часть кода: 

for fname in images:
    img = cv2.imread(fname)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    # Find the chess board corners
    # If desired number of corners are found in the image then ret = true
    ret, corners = cv2.findChessboardCorners(gray, CHECKERBOARD,
                                             cv2.CALIB_CB_ADAPTIVE_THRESH + cv2.CALIB_CB_FAST_CHECK + cv2.CALIB_CB_NORMALIZE_IMAGE)

    if ret == True:
        objpoints.append(objp)
        # refining pixel coordinates for given 2d points.
        corners2 = cv2.cornerSubPix(gray, corners, (11, 11), (-1, -1), criteria)

        imgpoints.append(corners2)

        # Draw and display the corners
        img = cv2.drawChessboardCorners(img, CHECKERBOARD, corners2, ret)

    cv2.imshow('img', img)
    cv2.waitKey(1)

cv2.destroyAllWindows()

h, w = img.shape[:2]

ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(objpoints, imgpoints, gray.shape[::-1], None, None)
...