小编Lau*_*a_M的帖子

如何在视频上正确运行 Detectron2?

我目前正在使用 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)

python opencv google-colaboratory

3
推荐指数
1
解决办法
2291
查看次数

标签 统计

google-colaboratory ×1

opencv ×1

python ×1