利用 MMDetection 处理视频

---

前言

在参加 OpenMMLab AI 实战营中，有个作业要求利用 MMDetection （目标检测工具包）训练模型，然后利用这个模型对一段小视频进行处理 （保留 mask 部分的彩色， 背景部分取灰）。文章的素材当然也来自实战营啦。这里主要讲一下处理视频的过程，训练模型的过程略过。

---

一、模型引入

from mmdet.apis import init_detector
model = init_detector(config="../work_dir/balloon.py", checkpoint="../work_dir/latest.pth")

这里 config 为训练模型的配置文件，checkpoint 为训练完保存的模型文件

---

二、查看图片

要利用 MMDetection 处理视频，就要先知道如何查看模型处理完之后的图片

from mmdet.apis import inference_detector, show_result_pyplot
img_path = "../data/balloon/val/3825919971_93fb1ec581_b.jpg"
result = inference_detector(model, img_path)
show_result_pyplot(model, img_path, result, score_thr=0.85)

结果如下：

---

三、视频每帧

我们处理视频，很容易可以想到讲视频拆穿一帧一帧处理，一帧就是一张图片，这里利用 mmcv 对视频进行处理