mmaction2 指南 (2) Finetune模型

mmaction2 指南

2. Finetunning 模型

Step 1: 定义数据集

Step 2: 修改配置

修改任务头 Head

2D/3D模型，backbone配置，分类头/定位头/检测头的配置（TSN，TSM，BSN）

model = dict(
    type='Recognizer2D',
    backbone=dict(
        type='ResNet',
        pretrained='torchvision://resnet50',
        depth=50,
        norm_eval=False),
    cls_head=dict(
        type='TSNHead',
        num_classes=101,   # change from 400 to 101
        in_channels=2048,
        spatial_type='avg',
        consensus=dict(type='AvgConsensus', dim=1),
        dropout_ratio=0.4,
        init_std=0.01))

修改数据集

配置一下数据集类型（视频/图片），数据根目录root，数据标注文件（文件路径默认前缀为数据根目