定义遮罩将在其中起作用的视频区域

Question

我有一个脚本，它根据遮罩检测到屏幕上的某些对象，但是我有一个感兴趣的区域，或者我只想在视频的某个区域应用遮罩，这是一个示例图片：

下面是我的脚本：

import cv2
import numpy as np
#Python + Open CV - Detecção de Fogo - Script Desenvolvido por Matheus Arruda
#0 -> busca o video da webcam // "nome_do_arquivo.mp4" faz a detecção com base em um video
video = cv2.VideoCapture("video.mp4")

while True:
    (grabbed, frame) = video.read()
    if not grabbed:
        break
    #filtro de exibição (blur)
    blur = cv2.GaussianBlur(frame, (5, 5), 0)
    #Metodo de detecção de cor (pode ser usado HSV ou RGB) RGB se utiliza o metodo padrão de cor, exemplo: Vermelho 255,0,0
    hsv = cv2.cvtColor(blur, cv2.COLOR_BGR2HSV)
    #Parametros de detecção de cor 
    minimo = [18, 70, 50]#[18, 50, 50]
    maximo = [35, 255, 255] #[35, 255, 255]
    minimo = np.array(minimo, dtype="uint8")
    maximo = np.array(maximo, dtype="uint8")
    #Na mascara, seto o tipo de filtro que nesse caso é o HSV e os ranges de detecção minimo e máximo
    mascara = cv2.inRange(hsv, minimo, maximo)
    #Desenho o contorno dos objetos detectados
    _, contours, _ = cv2.findContours(mascara, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
    for contour in contours:
        cv2.drawContours(frame, contour, -1, (0, 255, 0), 3)


    #Saida, frame = video, aplicado HSV e a mascara definida acima
    output = cv2.bitwise_and(frame, hsv, mask=mascara)
    #Conta a quantidade de pixels que não estão com vermelho, com base nessa informação geramos a condição de detectação de fogo
    px_sem_vermelho = cv2.countNonZero(mascara)
    font = cv2.FONT_HERSHEY_SIMPLEX
    if int(px_sem_vermelho) > 1500: #Padrão de detecção (20.000)
        print ('Detectado Fogo')
        cv2.putText(frame,'Vazando Ferro Liquido',(10,330), font, 0.7,(255,255,255),2,cv2.LINE_AA)
        print(px_sem_vermelho)
    elif int(px_sem_vermelho) > 900:
        print('Sem Fogo')
        cv2.putText(frame,'Vazando Pouco Ferro Liquido',(10,330), font, 0.7,(255,255,255),2,cv2.LINE_AA)
        print(int(px_sem_vermelho))
    else:
        print('Sem Fogo')       
        print(int(px_sem_vermelho))

    cv2.imshow("Filtro Deteccao", cv2.resize(output,(640,480)))
    cv2.imshow("Video", cv2.resize(frame,(640,480)))
    #print("output:", frame)

        #print("output:".format(mascara))
        #Tecla 'q' definida para sair do programa
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cv2.destroyAllWindows()
video.release()

Answer 1

似乎算法的分割/检测部分与特定的图像尺寸无关。因此，您可以使用感兴趣区域选择图像中感兴趣的部分，然后在此ROI上运行算法并将结果合并回去（这是可以完成的，因为您知道ROI的位置）

为了提取投资回报率，您可以使用简单的索引编制：

roi = image[ystart:yend][xstart:xend]

现在，您可以将roi用作任何图像。在其上运行算法，然后使用简单的分配更改image中的相应部分：

image[ystart:yend][xstart:xend] = processedROI