我想通过haar级联来计算汽车数量。
#import libraries of python opencv
import numpy as np
import cv2
import gc
import uuid
import datetime
import time
import csv
cap = cv2.VideoCapture('v3.mp4')
car_cascade = cv2.CascadeClassifier('cars.xml')
W = cap.get(3)
H = cap.get(4)
areaTH = 700
H1 = (H/2)+10
W1 = W/2
mx = 0
my = 30
while (cap.isOpened()):
#capture frame by frame
ret, frame = cap.read()
#convert video into gray scale of each frames
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
#detect cars in the video
cars = car_cascade.detectMultiScale(gray, 1.1, 3)
#to draw arectangle in each cars
for (x,y,w,h) in cars:
cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)
#display the resulting frame
cv2.imshow('video', frame)
#press Q on keyboard to exit
if cv2.waitKey(25) & 0xFF == ord('q'):
break
Line1 = np.array([[20,H1],[310,H1]], np.int32).reshape((-1,1,2))
frame = cv2.polylines(frame,[Line1],False,(0,0,255),thickness=5)
fram, contours0, hierarchy =
cv2.findContours(frame,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
for cnt in contours0:
# cv2.drawContours(frame, cnt, -1, (0,255,0), 2, 8)
area = cv2.contourArea(cnt)
#print ('Area : '+str(area))
if area > areaTH:
#################
# TRACKING #
#################
M = cv2.moments(cnt)
cx = int(M['m10']/M['m00'])
cy = int(M['m01']/M['m00'])
x,y,w,h = cv2.boundingRect(cnt)
cv2.circle(frame,(cx,cy), 3, (255,0,0), -1)
img = cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)
#print ('H1 : '+str(H1))
print('cy : '+str(cy))
if (cy >= 147) and (cy<= 155):
Vehicles = Vehicles + 1
Line1 = np.array([[200,H1],[880,H1]],
np.int32).reshape((-1,1,2))
frame = cv2.polylines(frame,[Line1],False,(255,0,0),thickness=5)
cv2.imshow('Frame',frame)
#Abort and exit with 'Q' or ESC
k = cv2.waitKey(30) & 0xff
if k == 27:
break
cap.release() #release video file
cv2.destroyAllWindows()
我将图像正确转换为灰度并成功在框架上画了一条线,问题是我遇到了这个错误
错误:OpenCV(3.4.2)C:\ build \ 3_4_winpack-bindings-win32-vc14-static \ opencv \ modules \ imgproc \ src \ contours.cpp:199:错误:(-210:不支持的格式或组合格式)[开始]在模式!= CV_RETR_FLOODFILL时,FindContours仅支持CV_8UC1图像,否则仅在函数'cvStartFindContours_Impl'中支持CV_32SC1图像
当我在此代码行中传递变量'frame'时,有人可以帮我解决这个错误,而不是x
fram, contours0, hierarchy =
cv2.findContours(frame,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
答案 0 :(得分:0)
如果您查看OpenCV文档,则会发现函数cv2.findContour()
需要单色图像。您通过了frame
,该频道仍处于3个频道中。您应该改用gray
变量(cv2.findContours(gray,...)
。
答案 1 :(得分:0)
从documentation到cv2.findContour()
图像 – 来源,一个8位单通道图像。非零像素视为1。零像素仍为0,因此图像被视为binary。您可以使用compare(),inRange(),threshold(),adaptiveThreshold(),Canny()和其他图像来创建灰度或彩色的二进制图像。该功能在提取轮廓时修改图像。如果mode等于CV_RETR_CCOMP或CV_RETR_FLOODFILL,则输入也可以是标签(CV_32SC1)的32位整数图像。
尝试使用此行缩放像素
image = cv2.convertScaleAbs(image)