使用Python中的openCV進行紅色和黃色三角形檢測

我正在嘗試使用Python中的openCV來檢測紅色三角形和黃色三角形。我是一名初學者。使用Python中的openCV進行紅色和黃色三角形檢測

我想，一方面，檢測，計數（黃色和紅色），並用矩形標出相機可以看到的所有三角形。我也想找到他們的羣衆中心。

目前，我只是一次檢測到一個單一的三角形而沒有發現它的顏色。我的質心的演算不能正常工作，給我的錯誤：

centroid_x = int(M['m10']/M['m00']) 
ZeroDivisionError: float division by zero

我已經寫了從實例靈感來自網絡下面的代碼

import numpy as np 
import cv2 

cap = cv2.VideoCapture(0) 

print cap.get(3) 
print cap.get(4) 

# changing display size 
ret = cap.set(3,320) 
ret = cap.set(4,240) 


def getthresholdedimg(hsv): 
    yellow = cv2.inRange(hsv,np.array((10,100,100)),np.array((30,255,255))) 
    red = cv2.inRange(hsv,np.array((0,0,0)),np.array((190,255,255))) 
    both = cv2.add(yellow,red) 
    return both 


def nothing(x): 
    pass 

# Create a black image, a window 
img = np.zeros((300,512,3), np.uint8) 
cv2.namedWindow('image') 

while(True): 
    thr1 = 50 
    thr2 = 110 


    # Capture frame-by-frame 
    ret, frame = cap.read() 

    # Our operations on the frame come here 
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) 

    gaussian_blur = cv2.GaussianBlur(gray,(5,5),0) 

    # Our operations on the frame come here 
    canny = cv2.Canny(gray,thr1,thr2) 
    canny_blur = cv2.Canny(gaussian_blur,thr1,thr2) 


    # Our operations on the frame come here 
    contours,hier = cv2.findContours(canny,1,2) 
    for cnt in contours: 
     approx = cv2.approxPolyDP(cnt,0.02*cv2.arcLength(cnt,True),True) 
     if len(approx)==3: 
      cv2.drawContours(frame,[cnt],0,(0,255,0),2) 
      tri = approx 

    M = cv2.moments(cnt) 
    centroid_x = int(M['m10']/M['m00']) 
    centroid_y = int(M['m01']/M['m00']) 
    cv2.circle(img,(centroid_x,centroid_y),3,255,-1) 

    for vertex in tri: 
     cv2.circle(frame,(vertex[0][0],vertex[0][1]),3,(64,0,128),-1) 
     cv2.line(img,(vertex[0][0],vertex[0][1]),(centroid_x,centroid_y),(0,0,255),1) 


    # Display the resulting frame 
    cv2.imshow('normal flux',frame) 
    cv2.imshow('gray conversion',gray) 
    cv2.imshow('canny edges conversion',canny) 
    cv2.imshow('canny edges gaussian blur',canny_blur) 

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

# When everything done, release the capture 
cap.release() 
cv2.destroyAllWindows()

你能幫助我嗎？

來源

2013-12-22 user3127579

也許你想要做 M = cv2.moments(tri) 而不是M = cv2.moments(cnt)？

來源

2013-12-22 17:50:28 smeso

您的相機可能沒有讀取它。

while(1): 

#Gets retval and frames from each video 

ret ,frame = cap.read() 

#Check to see if retval is not None or empty 
if not ret 
    break;

：你的帽子= cv2.videoCapture（0）之後

試試這個

來源

2015-04-01 22:51:35 OpenCv

使用Python中的openCV進行紅色和黃色三角形檢測

回答

相關問題