特定の色を検出したらウェブカメラを停止する

Question

ライトピンクの色を検出するコードを書いています。今度は、ライトピンクの色を検出した後、ウェブカメラを自動的に閉じるコードを追加したいと思います。これで私を助けてくれますか？それは意味的に何のループではありませんので、

import cv2 
import numpy as np 

cap = cv2.VideoCapture(0) 

while(1): 
    _, frame = cap.read() 

    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) 

    lower_pink = np.array([160,50,50]) 
    upper_pink = np.array([180,255,255]) 

    mask = cv2.inRange(hsv, lower_pink, upper_pink) 

    # Bitwise-AND mask and original image 
    res = cv2.bitwise_and(frame,frame, mask= mask) 
    cv2.imshow('frame',frame) 
    cv2.imshow('mask',mask) 
    cv2.imshow('res',res) 
    break 

if(cv2.countNonZero(mask) > 0): 
    print("FOUND") 
    raise SystemExit 

cv2.destroyAllWindows() 

Answer 1

無条件break（ノー可能continue）とのループは意味がありません： はここEDITEDコードです。

テストがループ内にある必要があります。最初に十分なピンクが付いた状態になるまで、キャプチャした各イメージに適用したいからです。そして、breakループ。ループの後のクリーンアップコードはもう実行されないので、ここでプログラムを終了しないでください。とにかくSystemExitを呼び出すことによって終了することは、ちょっと変わっています。それはsys.exit()の機能です。

import cv2 
import numpy as np 

def main(): 
    lower_pink = np.array([160, 50, 50]) 
    upper_pink = np.array([180, 255, 255]) 
    threshold = 100 # TODO Adapt to your needs. 
    cap = cv2.VideoCapture(0) 

    while True: 
     _, frame = cap.read() 
     hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) 
     mask = cv2.inRange(hsv, lower_pink, upper_pink) 
     masked = cv2.bitwise_and(frame, frame, mask=mask) 
     cv2.imshow('frame', frame) 
     cv2.imshow('mask', mask) 
     cv2.imshow('masked', masked) 
     # if cv2.countNonZero(mask) > threshold: 
     #  print('FOUND') 
     #  break 
     print(cv2.countNonZero(mask)) 
     # 
     # Wait for escape key. 
     # 
     if cv2.waitKey(500) == 27: 
      break 

    cv2.destroyAllWindows() 


if __name__ == '__main__': 
    main() 

実際の閾値テストは、あなたのニーズに合うと思われる値を決定できるように、マスクの画素数を印刷してコメントアウトして交換されます。