Question

我写了一个检测淡粉色的代码。现在我想添加一个代码，它会在检测到浅粉红色后自动关闭网络摄像头。你能帮帮我这个吗？这是 EDITED 代码：

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）的循环没有意义，因为它在语义上没有循环。

测试必须在循环内，因为你想将它应用于每个捕获的图像，直到你用足够的粉红色击中第一个。然后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()

实际的阈值测试被注释掉，并通过打印蒙版的像素数来代替，以便您可以确定哪个值适合您的需要。

检测到特定颜色后停止网络摄像头

1 个答案: