我目前正在使用 Detectron2 来处理视频中的人物检测,我一直在尝试运行以下代码来读取视频文件,逐帧进行预测并使用处理过的帧录制视频,但我得到了一个空视频文件。我为此创建的环境位于 Colaboratory 中,并具有以下版本(python 3.6、opencv 4.2.30)。我是新手,但如果有人能给我一个想法,我将不胜感激
这是代码
#!/usr/bin/env python3
# -- coding: utf-8 --
import detectron2
from detectron2.utils.logger import setup_logger
setup_logger()
# import some common libraries
import numpy as np
import cv2
# import some common detectron2 utilities
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg
from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog
import time
cap = cv2.VideoCapture('piloto legger 1.mp4')
hasFrame, frame = cap.read()
FPS = cap.get(cv2.CAP_PROP_FPS)
frame_width = frame.shape[1]
frame_height = frame.shape[0]
video_writer …Run Code Online (Sandbox Code Playgroud)