slowfast检测结果获取

slowfast检测结果获取

在成功运行完示例之后，想要根据检测结果进行后续工作，一个必不可少的数据便是模型的检测结果。在slowfast项目中，我暂时没找到官方提供的可用接口可以直接使用，在github中有看到类似信息，根据他的提示，改写部分代码，成功获取模型检测的分类结果。

1. 在slowfast官方项目中找到video_visualizer.py文件,slowfast->slowfast->visualization->video_visualizer.py

2.在该文件中找到draw_one_frame（）函数

该函数中的preds参数即为模型检测的结果，其值包含ava.json中所有类型的概率值。可以在这个函数中调用预测结果来实现你想要的功能。下图为我测试过程中的某画面的检测值，一共80个，一一对应ava.json文件中的类型。

当然，我们一般不需要这么多数据，我们关心的是最大概率值及其类别，或者说topk的值。这里，可以使用preds.topk(k=x),这里x即为所需要的top类型个数。
下图是输出的preds.topk(k=5)的结果，包含概率最大的5种类别以及对应的概率值。