ICCV | 2021 OadTR: Online Action Detection with Transformers代码复现遇到的问题

该论文提出了一种基于transformer的在线行为检测框架OadTR,该框架能够从视频流中识别出正在发生的行为，该模型可用在在自动驾驶，视频监控，异常检测等方面。

论文地址
代码地址

1.缺少安装包sklearn
解决：安装sklearn pip install sklearn
sklearn是一个Python第三方提供的非常强力的机器学习库，它包含了从数据预处理到训练模型的各个方面。sklearn拥有可以用于监督和无监督学习的方法，sklearn中的大部分函数可以归为估计器(Estimator)和转化器(Transformer)两类。估计器(Estimator)其实就是模型，它用于对数据的预测或回归。
基本上估计器都会有以下几个方法：
fit(x,y):传入数据以及标签即可训练模型，训练的时间和参数设置，数据集大小以及数据本身的特点有关
score(x,y)用于对模型的正确率进行评分(范围0-1)。
predict(x)用于对数据的预测，它接受输入，并输出预测标签，输出的格式为numpy数组。
转化器(Transformer)用于对数据的处理，例如标准化、降维以及特征选择等等。同与估计器的使用方法类似:
fit(x,y) :该方法接受输入和标签，计算出数据变换的方式。
transform(x) :根据已经计算出的变换方式，返回对输入数据x变换后的结果.

2.CUDA 和pytorch版本不匹配
可以从该网址中查看对应的pytorch的版本的cuda
该处的pytorch版本为1.6.
我的解决办法是先更新自己的显卡驱动，
ubuntu-drivers devices（查看自己的显卡）
$ sudo ubuntu-drivers autoinstall1（开始自动安装）
完成后重启
sudo add-apt-repository ppa:graphics-drivers/ppa（将 ppa:graphics-drivers/ppa 存储库添加到系统中）
sudo apt-get update（更新系统）
打开软件与更新 ——》 选择附加驱动 ——》 选择要应用的驱动版本 ——》点击应用更改（找到对应的版本）（参考自知乎链接https://www.zhihu.com/question/309583980）
根据自己的驱动，选择对应的CUDA版本，安装pytorch

3.找不到thumos_val_anno.pickle文件
解决办法：修改dataset.py文件中的self.pickle_root的路径为anno_thumos所在的地方。

4.找不到Anet2016_feature_v2/video_validation_0000690_rgb.npy’文件夹
No such file or directory: '/home/ubuntu/project/OadTR/OadTR-main-code/data/OadTR_THUMOS/Feature/Anet2016_feature_v2/video_validation_0000690_rgb.npy
解决办法：将下载下来的特征文件解压后，放在pickle_root的路径下面