Python에서 openCV를 사용하는 빨간색과 노란색 삼각형 탐지

Question

python에서 openCV를 사용하여 빨간색 삼각형과 노란색 삼각형을 구별하려고합니다. 나는 초보자이다.

처음에는 카메라가 볼 수있는 모든 삼각형을 감지하고, (노란색과 빨간색으로) 계산하고 사각형으로 표시하고 싶습니다. 또한 그들의 대중 센터를 찾고 싶습니다.

현재로서는 색상을 찾지 않고 한 번에 하나의 단일 삼각형 만 감지합니다.

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() 

당신이 날 도와 줘요 수 있습니까 웹에서 예로부터 영감을 다음과 같은 코드를 작성했다

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

: 질량 중심의 내 calcul 나에게 오류를주고, 작동하지 않는 이유는 무엇입니까?

Answer 1

M = cv2.moments(tri) M = cv2.moments(cnt) 대신에?

Answer 2

카메라가 사진을 읽지 못하고 있습니다.

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) 후이 시도