OpenCV+Python跟踪指定颜色

OpenCV简易视频处理框架
OpenCV主要色彩空间
OpenCV的位操作方法

找出视频中红色物体

# image_processing.py
def filter_out_red(src_frame):
    if src_frame is not None:
        hsv = cv2.cvtColor(src_frame, cv2.COLOR_BGR2HSV)
        lower_red = np.array([156, 43, 46])
        upper_red = np.array([180, 255, 255])
        # inRange()方法返回的矩阵只包含0,255 (CV_8U) 0表示不在区间内
        mask = cv2.inRange(hsv, lower_red, upper_red)
        return cv2.bitwise_and(src_frame, src_frame, mask=mask)

# scratch.py
def run(self):
    self._window_manager.create_window()
    while self._window_manager.is_window_created:
        self._capture_manager.enter_frame()
        frame = self._capture_manager.frame
        if frame is not None:
            self._capture_manager.frame = image_processing.filter_out_red(frame)
        self._capture_manager.exit_frame()
        self._window_manager.process_events()

原始视频

过滤出红色部分.gif