docs/zh_cn/user_guides/inference.md

# 使用现有模型进行推理

MMAction2 在[模型库](../modelzoo.md)中提供了预训练的视频理解模型。本文将展示如何使用现有模型对给定的视频进行推理。

关于如何在标准数据集上测试现有模型，请参考这个[指南](./train_test.md#test)。

## 对给定视频进行推理

MMAction2 提供了用于对给定视频进行推理的高级 Python API：

- [init_recognizer](mmaction.apis.init_recognizer): 使用配置文件和权重文件初始化一个识别器
- [inference_recognizer](mmaction.apis.inference_recognizer): 对给定视频进行推理

下面是一个使用 Kinitics-400 预训练权重构建模型并对给定视频进行推理的示例。

```{note}

如果您将 mmaction2 用作第三方包，您需要下载示例中的配置文件和演示视频。



运行 'mim download mmaction2 --config tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb --dest .' 下载所需的配置文件。



运行 'wget https://github.com/open-mmlab/mmaction2/blob/main/demo/demo.mp4' 下载所需的演示视频。

```

```python

from mmaction.apis import inference_recognizer, init_recognizer



config_path = 'configs/recognition/tsn/tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb.py'

checkpoint_path = 'https://download.openmmlab.com/mmaction/v1.0/recognition/tsn/tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb/tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb_20220906-2692d16c.pth' # 可以是本地路径

img_path = 'demo/demo.mp4'   # 您可以指定自己的图片路径



# 从配置文件和权重文件中构建模型

model = init_recognizer(config_path, checkpoint_path, device="cpu")  # device 可以是 'cuda:0'

# 对单个视频进行测试

result = inference_recognizer(model, img_path)

```

`result` 是一个包含 `pred_scores` 的字典。

示例中的动作识别演示可以在[demo/demo.py](https://github.com/open-mmlab/mmaction2/blob/main/demo/demo.py)中找到。