Skip to content

Latest commit

 

History

History
157 lines (107 loc) · 7.46 KB

pp-timesformer.md

File metadata and controls

157 lines (107 loc) · 7.46 KB
Raw

English | 简体中文

PP-TimeSformer视频分类模型

内容

模型简介

我们对TimeSformer模型进行了改进和优化，得到了更高精度的2D实用视频分类模型PP-TimeSformer。在不增加参数量和计算量的情况下，在UCF-101、Kinetics-400等数据集上精度显著超过原版，在Kinetics-400数据集上的精度如下表所示。

Version Top1
Ours (swa+distill+16frame) 79.44
Ours (swa+distill) 78.87
Ours (swa) 78.61
mmaction2 77.92

数据准备

K400数据下载及准备请参考Kinetics-400数据准备

UCF101数据下载及准备请参考UCF-101数据准备

模型训练

Kinetics-400数据集训练

下载并添加预训练模型

  1. 下载图像预训练模型ViT_base_patch16_224_miil_21k.pdparams作为Backbone初始化参数，或通过wget命令下载

    wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ViT_base_patch16_224_pretrained.pdparams

  2. 打开PaddleVideo/configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml，将下载好的权重存放路径填写到下方pretrained:之后

    MODEL:
    framework: "RecognizerTransformer"
    backbone:
        name: "VisionTransformer"
        pretrained: 将路径填写到此处

开始训练

  • Kinetics400数据集使用8卡训练，训练方式的启动命令如下:

    # videos数据格式
python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3,4,5,6,7"  --log_dir=log_pptimesformer  main.py  --validate -c configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml

  • 开启amp混合精度训练，可加速训练过程，其训练启动命令如下：

    export FLAGS_conv_workspace_size_limit=800 # MB
export FLAGS_cudnn_exhaustive_search=1
export FLAGS_cudnn_batchnorm_spatial_persistent=1
# videos数据格式
python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3,4,5,6,7"  --log_dir=log_pptimesformer  main.py --amp --validate -c configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml

  • 另外您可以自定义修改参数配置，以达到在不同的数据集上进行训练/测试的目的，建议配置文件的命名方式为模型_数据集名称_文件格式_数据格式_采样方式.yaml，参数用法请参考config。

模型测试

  • PP-TimeSformer模型在训练时同步进行验证，您可以通过在训练日志中查找关键字best获取模型测试精度，日志示例如下:

    Already save the best model (top1 acc)0.7258

  • 由于PP-TimeSformer模型测试模式的采样方式是速度稍慢但精度高一些的UniformCrop，与训练过程中验证模式采用的RandomCrop不同，所以训练日志中记录的验证指标topk Acc不代表最终的测试分数，因此在训练完成之后可以用测试模式对最好的模型进行测试获取最终的指标，命令如下：

    # 8-frames 模型测试命令
python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3,4,5,6,7"  --log_dir=log_pptimesformer  main.py  --test -c configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml -w "output/ppTimeSformer/ppTimeSformer_best.pdparams"

# 16-frames模型测试命令
python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3,4,5,6,7"  --log_dir=log_pptimesformer main.py --test \
-c configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml \
-o MODEL.backbone.num_seg=16 \
-o MODEL.runtime_cfg.test.num_seg=16 \
-o MODEL.runtime_cfg.test.avg_type='prob' \
-o PIPELINE.test.decode.num_seg=16 \
-o PIPELINE.test.sample.num_seg=16 \
-w "data/ppTimeSformer_k400_16f_distill.pdparams"

    当测试配置采用如下参数时，在Kinetics-400的validation数据集上的测试指标如下：

    backbone Sampling method num_seg target_size Top-1 checkpoints
    Vision Transformer UniformCrop 8 224 78.61 ppTimeSformer_k400_8f.pdparams
    Vision Transformer UniformCrop 8 224 78.87 ppTimeSformer_k400_8f_distill.pdparams
    Vision Transformer UniformCrop 16 224 79.44 ppTimeSformer_k400_16f_distill.pdparams

  • 测试时，PP-TimeSformer视频采样策略为使用linspace采样：时序上，从待采样视频序列的第一帧到最后一帧区间内，均匀生成num_seg个稀疏采样点（包括端点）；空间上，选择长边两端及中间位置（左中右 或 上中下）3个区域采样。1个视频共采样1个clip。

模型推理

导出inference模型

python3.7 tools/export_model.py -c configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml \
                                -p data/ppTimeSformer_k400_8f.pdparams \
                                -o inference/ppTimeSformer

上述命令将生成预测所需的模型结构文件ppTimeSformer.pdmodel和模型权重文件ppTimeSformer.pdiparams。

使用预测引擎推理

python3.7 tools/predict.py --input_file data/example.avi \
                           --config configs/recognition/pptimesformer/pptimesformer_k400_videos.yaml \
                           --model_file inference/ppTimeSformer/ppTimeSformer.pdmodel \
                           --params_file inference/ppTimeSformer/ppTimeSformer.pdiparams \
                           --use_gpu=True \
                           --use_tensorrt=False

输出示例如下:

Current video file: data/example.avi
        top-1 class: 5
        top-1 score: 0.9997474551200867

可以看到，使用在Kinetics-400上训练好的ppTimeSformer模型对data/example.avi进行预测，输出的top1类别id为5，置信度为0.99。通过查阅类别id与名称对应表data/k400/Kinetics-400_label_list.txt，可知预测类别名称为archery。

参考论文