¿Cómo dibujar un rectángulo relleno en cada cuadro de video usando Python-OpenCV?

En este artículo, discutiremos cómo dibujar un rectángulo relleno en cada cuadro de video a través de OpenCV en Python.

Implementación paso a paso:

• Importe las bibliotecas necesarias en el espacio de trabajo.
• Lee el video en el que tienes que escribir.

Sintaxis:

cap = cv2.VideoCapture(“ruta”)

• Cree un archivo de salida usando el método cv2.VideoWriter_fourcc(). Aquí tendrás opciones de formatos de video.

Sintaxis:

salida = cv2.VideoWriter(“salida.avi”, cv2.VideoWriter_fourcc(*’MPEG’), 30, (1080, 1920))

• Luego, edite los cuadros del video agregándole formas (en nuestro caso, es un rectángulo relleno). Usaremos el método cv2.rectangle() . Este método se utiliza para dibujar un rectángulo en cualquier imagen.

Sintaxis:

cv2.rectangle(marco, (100, 100), (500, 500), (0, 255, 0), -1)

• Luego escribe todos los cuadros en el archivo de video.

Sintaxis:

salida.escribir(marco)

Ejemplo:

En este ejemplo, agregamos un rectángulo verde al video.

Vídeo de entrada:

import cv2
  
  
def main():
    
    # reading the input
    cap = cv2.VideoCapture("input.mp4")
  
    output = cv2.VideoWriter(
        "output.avi", cv2.VideoWriter_fourcc(*'MPEG'), 
      30, (1080, 1920))
  
    while(True):
        ret, frame = cap.read()
        if(ret):
              
            # adding filled rectangle on each frame
            cv2.rectangle(frame, (100, 150), (500, 600),
                          (0, 255, 0), -1)
              
            # writing the new frame in output
            output.write(frame)
            cv2.imshow("output", frame)
            if cv2.waitKey(1) & 0xFF == ord('s'):
                break
        else:
            break
  
    cv2.destroyAllWindows()
    output.release()
    cap.release()
  
  
if __name__ == "__main__":
    main()

Producción: