Python姿态估计示例误差

Question

我是opencv的初学者，我正在尝试运行教程代码： calib3d.html

我的代码和问题：

import scipy.io
import numpy as np
import cv2
import time

# data comes from calibration program same as tutorial
data=scipy.io.loadmat('distm.mat')
dist=data['dist000']
mtx=data['mtx']
data=None

#takes the corners in the chessboard  
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

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

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

# cam open and wait a second to deny black or dark images                                                                              
cap = cv2.VideoCapture(2)
ret, frame = cap.read()
print "Checking camera read:"+str(ret)
while ret ==False:
    cap = cv2.VideoCapture(2)
    ret, frame = cap.read()
time.sleep(1)

# takes live photos from camera
for i in range(0,20):
    print i

    ret, img = cap.read()
    gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    ret, corners = cv2.findChessboardCorners(gray, (7,6),None)
    print ret

    if ret == True:

        corners2 = cv2.cornerSubPix(gray,corners,(11,11),(-1,-1),criteria)
        # 'corners2' returns "none" and below codes not working so I use 'corners'
        print corners2
        # Find the rotation and translation vectors.
        rvecs, tvecs, inliers = cv2.solvePnPRansac(objp, corners, mtx, dist)

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

        img = draw(img,corners,imgpts)  # img returns None
        print img                       
        cv2.imshow('pose',img)           # error line 
        k = cv2.waitKey(0) & 0xff

，并在“cv2.imShow(‘姿态’，img)”处给出错误为：

OpenCV Error: Assertion failed (size.width>0 && size.height>0) in
cv::imshow, file ..\..\..\modules\highgui\src\window.cpp, line 269
Traceback (most recent call last):   File "<stdin>", line 1, in
<module>   File
"C:\Python27\lib\site-packages\spyderlib\widgets\externalshell\sitecustomize.py",
line 540, in runfile
    execfile(filename, namespace)   File "C:/Users/ACS/Documents/Python/Pose Estimation.py", line 49, in
<module>
    cv2.imshow('pose',img) cv2.error: ..\..\..\modules\highgui\src\window.cpp:269: error: (-215)
size.width>0 && size.height>0 in function cv::imshow

感谢您的关注。

Answer 1

有一件事可能会引起麻烦，那就是没有检查cap.read()的成功:变量ret被重新分配到一个新的值，而不需要第一次检查。

围绕cap.read()下面的代码使用另一个if ret:可能会有所帮助。您会注意到在Python中没有必要使用== True

Answer 2

谢谢阿诺的兴趣。

我解决了问题。工作代码：

import scipy.io
import numpy as np
import cv2
import time

data=scipy.io.loadmat('distm.mat')
dist=data['dist000']
mtx=data['mtx']
data=None

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

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

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

cap = cv2.VideoCapture(2)
ret, frame = cap.read()
print "Checking camera read:"+str(ret)
while ret ==False:
    cap = cv2.VideoCapture(2)
    ret, frame = cap.read()
time.sleep(1)

for i in range(0,20):
    time.sleep(1)
    ret, img = cap.read()
    gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    ret, corners = cv2.findChessboardCorners(gray, (7,6),None)
    print str(i)+"'th image check board corners "+str(ret)
    if ret :
        corners2 = cv2.cornerSubPix(gray,corners,(11,11),(-1,-1),criteria)
        print corners2
        # Find the rotation and translation vectors.
        rvecs, tvecs, inliers = cv2.solvePnPRansac(objp, corners, mtx, dist)

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

        img = draw(img,corners,imgpts)
        cv2.imshow('pose',img)
        k = cv2.waitKey(0) & 0xff