projectPoint не работает в openCV, выдавая ошибку на неверных аргументах - PullRequest
Новая
       20

projectPoint не работает в openCV, выдавая ошибку на неверных аргументах

0 голосов
/ 04 ноября 2018

Я использую OpenCV 3.1.0 с Anaconda и я пытался найти ориентацию объекта из 2D-изображения, и мой код заканчивался с ошибкой, уведомляющей, что

{{{Ошибка OpenCV: неверный аргумент (вращение должно быть представлено вектором вращения с плавающей точкой 1x3 или 3x1 или матрицей вращения 3x3) в файле cvProjectPoints2, файл / home / travis / miniconda / conda-bld / conda_1486587069159 / work / opencv-3.1.0 / modules / calib3d / src / calib.cpp, строка 608}}

вот мой код: 

def draw(img, corners, imgpts):
    corner = tuple(corners[0].ravel())
    img = cv2.line(img, corner, tuple(imgpts[0].ravel()), (255,0,0), 5)
    img = cv2.line(img, corner, tuple(imgpts[1].ravel()), (0,255,0), 5)
    img = cv2.line(img, corner, tuple(imgpts[2].ravel()), (0,0,255), 5)
    return img

cap = cv2.VideoCapture(1)

with np.load('calibData.npz') as X:
    mtx, dist, rvecs, tvecs = [X[i] for i in 
                                       ('mtx','dist','rvecs','tvecs')]

criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)
objp = np.zeros((7*7,3), np.float32)
objp[:,:2] = np.mgrid[0:7,0:7].T.reshape(-1,2)

axis = np.float32([[3,0,0], [0,3,0], [0,0,-3]]).reshape(-1,3)

while(1):
    ret1, frame = cap.read()
# print(frame)
    cv2.imshow('frame', frame)
    img = frame

    gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    ret, corners = cv2.findChessboardCorners(gray, (7,7),None)


    if ret == True:
        i += 1
        corners2 = cv2.cornerSubPix(gray,corners,(11,11),(-1,-1),criteria)

        # Find the rotation and translation vectors.
        rvecs, tvecs, inliers = cv2.solvePnP(objp, corners2, mtx, dist)

        # project 3D points to image plane
        imgpts, jac = cv2.projectPoints(axis, rvecs, tvecs, mtx, dist)

        img = draw(img,corners2,imgpts)
        cv2.imshow('img',img)
        cv2.waitKey(500)
        k = cv2.waitKey(0) & 0xff
        if k == 's':
            cv2.imwrite('fname'+str(i)+'.png', img)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

и вот сообщение об ошибке

Ошибка OpenCV: неверный аргумент (вращение должно быть представлено вектором вращения с плавающей точкой 1x3 или 3x1 или матрицей вращения 3x3) в файле cvProjectPoints2, файл /home/travis/miniconda/conda-bld/conda_1486587069159/work/opencv-3.1 .0 / modules / calib3d / src / calib.cpp, строка 608 Traceback (последний вызов был последним): Файл "pose.py", строка 45, в imgpts, jac = cv2.projectPoints (ось, rvecs, tvecs, mtx, dist) cv2.error: /home/travis/miniconda/conda-bld/conda_1486587069159/work/opencv-3.1.0/modules/calib3d/src/calibration.cpp:608: ошибка: (-5) Вращение должно быть представлено 1x3 или Вектор вращения с плавающей точкой 3x1 или матрица вращения 3x3 в функции cvProjectPoints2

...