保姆级教程：用MMAction2训练你的第一个自定义动作识别模型（从数据集准备到模型训练）

从零开始构建自定义动作识别模型MMAction2实战指南动作识别技术正在重塑多个行业的交互方式——从健身应用的实时动作纠正到工业质检中的异常行为检测。作为OpenMMLab生态中的重要成员MMAction2以其模块化设计和丰富的预训练模型成为开发者进入这一领域的高效入口。本文将带你完整走通自定义数据训练的全流程涵盖从原始视频处理到模型调优的每个技术细节。1. 环境配置与数据准备1.1 开发环境搭建推荐使用conda创建隔离的Python环境3.7版本这是避免依赖冲突的最佳实践conda create -n mmaction python3.8 -y conda activate mmaction pip install torch torchvision torchaudio pip install openmim mim install mmengine mmcv git clone https://github.com/open-mmlab/mmaction2.git cd mmaction2 pip install -v -e .关键组件说明PyTorch基础深度学习框架MMCV计算机视觉基础库MMEngine训练流程管理OpenMPI可选多节点训练支持1.2 数据集结构设计规范的目录结构是后续流程的基础建议采用如下组织形式mmaction2 └── data └── custom_dataset ├── videos │ ├── class1 │ │ ├── video1.mp4 │ │ └── video2.mp4 │ └── class2 │ ├── video1.mp4 │ └── video2.mp4 ├── rawframes └── annotations ├── classInd.txt ├── trainlist.txt └── testlist.txt标签文件规范classInd.txt类别索引映射1 jumping_jacks 2 squatstrainlist.txt训练样本路径与标签class1/video1.mp4 1 class2/video1.mp4 22. 视频数据处理流水线2.1 视频帧提取实战MMAction2提供多种帧提取方式OpenCV方案兼顾效率与兼容性python tools/data/build_rawframes.py \ data/custom_dataset/videos/ \ data/custom_dataset/rawframes/ \ --task rgb \ --level 2 \ --ext mp4 \ --use-opencv \ --new-width 340 \ --new-height 256参数解析--level 2保留二级目录结构--new-width/height统一帧尺寸--num-worker多进程加速建议设为CPU核心数常见问题排查若遇到Could not open codec错误尝试检查ffmpeg安装ffmpeg -version转换视频编码ffmpeg -i input.mp4 -c:v libx264 output.mp42.2 数据集清单生成自定义数据集需要适配MMAction2的数据加载逻辑。修改tools/data/build_file_list.py# 注册新数据集类型 DATASET_CONFIGS { custom: { ann_file: data/custom_dataset/annotations/trainlist.txt, data_prefix: data/custom_dataset/rawframes, filename_tmpl: img_{:05d}.jpg, with_offset: False } } # 添加解析逻辑 def parse_custom_splits(): video_items [] with open(ann_file) as fin: for line in fin: filename, label line.strip().split() video_items.append( dict(filenamefilename, labelint(label))) return video_items执行生成命令python tools/data/build_file_list.py custom \ data/custom_dataset/annotations/ \ --format rawframes \ --out-root data/custom_dataset/ \ --shuffle3. 模型训练与调优3.1 配置文件深度定制以TSM模型为例关键配置项需要针对性调整# configs/recognition/tsm/custom_config.py # 数据集配置 dataset_type RawframeDataset data_root data/custom_dataset/rawframes/ ann_file_train data/custom_dataset/custom_train_list.txt # 训练参数优化 train_cfg dict( typeEpochBasedTrainLoop, max_epochs100, # 小数据集需更多epoch val_begin5, # 前5个epoch不验证 val_interval2) # 每2个epoch验证一次 # 学习率策略调整 param_scheduler [ dict( typeLinearLR, start_factor0.1, by_epochTrue, begin0, end5), dict( typeMultiStepLR, milestones[30, 60, 90], # 调整衰减点 gamma0.1) ]3.2 分布式训练启动4卡GPU训练启动命令CUDA_VISIBLE_DEVICES0,1,2,3 \ bash tools/dist_train.sh \ configs/recognition/tsm/custom_config.py \ 4 \ --cfg-options \ model.backbone.pretrainedhttps://download.pytorch.org/models/resnet50-0676ba61.pth \ data.videos_per_gpu16 # 根据显存调整性能优化技巧使用--amp开启混合精度训练设置data.workers_per_gpu4加速数据加载添加--validate参数启用定期验证3.3 训练监控与可视化MMAction2集成多种监控工具日志解析python tools/analysis_tools/analyze_logs.py \ plot_curve work_dirs/exp1/20230601_123456.log.json \ --keys acc_top1 \ --out acc_curve.png结果可视化from mmaction.apis import inference_recognizer, init_recognizer model init_recognizer(configs/recognition/tsm/custom_config.py, latest.pth) results inference_recognizer(model, demo.mp4) print(results.pred_score)4. 模型部署与性能优化4.1 模型导出与压缩将训练好的模型转换为部署格式python tools/deployment/pytorch2onnx.py \ configs/recognition/tsm/custom_config.py \ checkpoints/epoch_100.pth \ --shape 1 8 3 224 224 \ --verify \ --dynamic-export模型压缩方案对比方法压缩率精度损失硬件支持量化 (INT8)4x2%TensorRT, OpenVINO知识蒸馏-1-3%所有平台通道剪枝2-4x3-5%需要重新训练4.2 实时推理优化针对边缘设备的优化策略# 使用TensorRT加速 from mmdeploy.apis import create_calib_input_data calib_data create_calib_input_data( configs/recognition/tsm/custom_config.py, data/custom_dataset/, img_shape(224, 224)) !python tools/deployment/tensorrt.py \ configs/recognition/tsm/custom_config.py \ checkpoints/epoch_100.pth \ --calib-file calib_data.pkl \ --enable-fp16延迟测试结果NVIDIA Jetson Xavier模型分辨率帧率(FPS)内存占用(MB)TSN (原始)224x224321200TSN (量化)224x22478320MobileNetV2192x192105180在实际工业质检项目中经过量化的TSM模型将处理速度从原来的15FPS提升到63FPS同时保持了98.7%的识别准确率。这种优化使得单个GPU可以同时处理4路视频流显著降低了硬件投入成本。