如何从视频中提取 ROI - 使不相关的像素变黑

Posted 2023-02-17

【中文标题】如何从视频中提取 ROI - 使不相关的像素变黑

【英文标题】：How do I extract ROI from video - blacken non relevant pixels

【发布时间】：2016-03-22 18:46:40

【问题描述】：

我正在用笔记本电脑的摄像头捕捉我移动头部的简单视频，并跟踪我的头部移动。

这是代码

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

# take first frame of the video
ret, frame = cap.read()


# setup initial location of window
r, h, c, w = 250, 90, 400, 125  # simply hardcoded the values
track_window = (c, r, w, h)

# set up the ROI for tracking
roi = frame[r:r + h, c:c + w]
hsv_roi = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv_roi, np.array((0., 60., 32.)), np.array((180., 255., 255.)))
roi_hist = cv2.calcHist([hsv_roi], [0], mask, [180], [0, 180])
cv2.normalize(roi_hist, roi_hist, 0, 255, cv2.NORM_MINMAX)

# Setup the termination criteria, either 10 iteration or move by atleas1t 1 pt
term_crit = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1)

while (1):
    ret, frame = cap.read()
    if ret:
        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
        dst = cv2.calcBackProject([hsv], [0], roi_hist, [0, 180], 1)
        # apply meanshift to get the new location
        ret, track_window = cv2.meanShift(dst, track_window, term_crit)
        # Draw it on image
        x, y, w, h = track_window
        cv2.rectangle(frame, (x, y), (x+w, y+h), (255,0, 0), 2)
        cv2.imshow('img2',frame)
        k = cv2.waitKey(60) & 0xff
        if k == 27:
            break
    else:
        break
cv2.destroyAllWindows()
cap.release()

我想将不在蓝色小矩形中的所有内容涂黑（也称为剪切）。

我尝试使用frame[:x,:y] = 0 和frame[x+w:,y+h:] = 0 使框架变黑，但我仍然看到矩形之外的部分不是黑色的。

有没有简单的方法可以做到这一点？

【问题讨论】：

什么是“节”？可以提供截图吗？

【参考方案1】：

这对我有用。也使用您的代码对其进行了测试。

# Draw it on image
    x, y, w, h = track_window
    cv2.rectangle(frame, (x, y), (x+w, y+h), (255,0, 0), 2)
    dst = np.zeros_like(frame)
    dst[y:y+h,x:x+w] = frame[y:y+h,x:x+w]
    cv2.imshow('dst',dst)