python - Python姿势估计示例错误

Question

我是opencv的初学者，我正在尝试在以下位置运行教程代码: http://docs.opencv.org/trunk/doc/py_tutorials/py_calib3d/py_table_of_contents_calib3d/py_table_of_contents_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('pose'，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