我知道这个问题还有其他帖子,但没有一个对我有帮助:
import cv2 as cv
import numpy as np
widthImg = 640
heightImg = 480
frameWidth = 640
frameHeight = 480
cap = cv.VideoCapture(2)
cap.set(3, widthImg)
cap.set(4, heightImg)
cap.set(10,150)
def preProcessing(img):
imgGray = cv.cvtColor(img,cv.COLOR_BGR2GRAY)
imgBlur = cv.GaussianBlur(imgGray,(5,5),1)
imgCanny = cv.Canny(imgBlur,200,200)
kernel = np.ones((5,5))
imgDial = cv.dilate(imgCanny,kernel,iterations=2)
imgThres = cv.erode(imgDial,kernel,iterations=1)
return imgThres
def getContours(img):
biggest = np.array([])
maxArea = 0
contours,hierarchy = cv.findContours(img,cv.RETR_EXTERNAL,cv.CHAIN_APPROX_NONE)
for cnt in contours:
area = cv.contourArea(cnt)
if area>5000:
#cv.drawContours(imgContour, cnt, -1, (255,0,0), 3)
peri = cv.arcLength(cnt,True)
approx = cv.approxPolyDP(cnt,0.02*peri,True)
if area >maxArea and len(approx) == 4:
biggest = approx
maxArea = area
cv.drawContours(imgContour, biggest, -1, (255, 0, 0), 20)
return biggest
def getWarp(img,biggest):
pts1 = np.array(biggest,np.float32)
pts2 = np.array([[0, 0], [widthImg, 0], [0, heightImg], [widthImg, heightImg]],np.float32)
matrix = cv.getPerspectiveTransform(pts1, pts2)
imgOutput = cv.warpPerspective(img, matrix, (widthImg, heightImg))
return imgOutput
while True:
success, img = cap.read()
img = cv.resize(img,(widthImg,heightImg))
imgContour = img.copy()
imgThres = preProcessing(img)
biggest = getContours(imgThres)
print(biggest)
imgWarped = getWarp(img,biggest)
cv.imshow("Result", imgWarped)
if cv.waitKey(1) & 0xFF == ord('q'):
break
这是我在Pycharm中的代码。我知道当我在44行使用“最大”时问题就来了(pts1=np.array(max,np.float32)。我知道如果我写4个二维点而不是“最大”,它会工作。但是在我下面的一个视频中,程序员使用“最大”没有任何问题。我知道它已经说过“最大”有4个点(在def getcours(img)中):).我不知道为什么在我和其他人的情况下不起作用是的
这是我得到的错误信息:
Traceback (most recent call last):
File "/home/nvidia/PycharmProjects/OpencvPython/Resources/Project2.py", line 61, in <module>
imgWarped = getWarp(img,biggest)
File "/home/nvidia/PycharmProjects/OpencvPython/Resources/Project2.py", line 46, in getWarp
matrix = cv.getPerspectiveTransform(pts1, pts2)
cv2.error: OpenCV(4.5.1) /tmp/pip-req-build-q3gzfcr4/opencv/modules/imgproc/src/imgwarp.cpp:3392: error: (-215:Assertion failed) src.checkVector(2, CV_32F) == 4 && dst.checkVector(2, CV_32F) == 4 in function 'getPerspectiveTransform'
抱歉,如果这是一个愚蠢的问题,但我是python编程新手,我花了几个小时试图解决它
用这个我就是这样解决的
相关问题 更多 >
编程相关推荐