使用mmdetection一帧一帧检测图片并拼接成视频

最新推荐文章于 2024-07-24 16:33:34 发布

clcshr1

最新推荐文章于 2024-07-24 16:33:34 发布

阅读量846

点赞数 1

本文链接：https://blog.youkuaiyun.com/clcshr1/article/details/116722179

版权

在demo文件夹下创建demo.py，使用mmdetection训练好的模型，对视频每一帧进行检测并画上bounding boxes，然后把这些帧拼接回一个视频。

import argparse
import numpy as np
import torch
from mmdet.apis import inference_detector, init_detector, show_result
import cv2
from moviepy.editor import VideoFileClip

def parse_args():
	#config和checpoint必填
    parser = argparse.ArgumentParser(description='MMDetection webcam demo')
    parser.add_argument('--config', help='test config file path',default="/home/liucs/mmdetection/config_clc/faster_rcnn_r101_fpn_1x.py")
    parser.add_argument('--checkpoint', help='checkpoint file',default="/home/liucs/mmdetection/work_dirs/clc/epoch_12.pth")
    parser.add_argument('--device', type=int, default=0, help='CUDA device id')
    parser.add_argument(
        '--camera-id', type=int, default=0, help