当前位置：首页 > news >正文

终极指南：如何用3D-ResNets-PyTorch快速搭建动作识别模型

news 2026/4/20 3:24:03

终极指南：如何用3D-ResNets-PyTorch快速搭建动作识别模型

【免费下载链接】3D-ResNets-PyTorch3D ResNets for Action Recognition (CVPR 2018)项目地址: https://gitcode.com/gh_mirrors/3d/3D-ResNets-PyTorch

3D-ResNets-PyTorch是一个基于PyTorch实现的动作识别框架，它利用3D ResNet架构来提取视频中的时空特征，实现高精度的动作识别。本指南将带你从零开始，快速掌握使用这个强大工具构建动作识别模型的完整流程。

📋 准备工作：环境搭建与依赖安装

在开始之前，我们需要确保你的系统满足以下要求：

Python 3.x
PyTorch 0.4+
FFmpeg和FFprobe（用于视频处理）

一键安装核心依赖

使用conda命令可以快速安装PyTorch：

conda install pytorch torchvision cudatoolkit=10.1 -c soumith

获取项目代码

通过以下命令克隆项目仓库：

git clone https://gitcode.com/gh_mirrors/3d/3D-ResNets-PyTorch

🎥 数据准备：从视频到训练数据

动作识别需要处理大量视频数据，3D-ResNets-PyTorch支持多种主流数据集，包括Kinetics、UCF-101、HMDB-51和ActivityNet。下面以Kinetics为例，展示完整的数据准备流程。

1. 视频下载

使用官方提供的爬虫工具下载Kinetics数据集：

# 参考官方爬虫：https://github.com/activitynet/ActivityNet/tree/master/Crawler/Kinetics

2. 视频转图片序列

将视频文件转换为jpg图片序列，使用项目提供的工具脚本：

python -m util_scripts.generate_video_jpgs mp4_video_dir_path jpg_video_dir_path kinetics

3. 生成标注文件

将原始标注转换为模型所需的json格式：

python -m util_scripts.kinetics_json csv_dir_path 700 jpg_video_dir_path jpg dst_json_path

🚀 训练你的第一个3D ResNet模型

一切准备就绪后，我们可以开始训练模型了。项目提供了直观的命令行接口，让训练过程变得简单。

基础训练命令

训练ResNet-50模型在Kinetics-700数据集上的基本命令：

python main.py --root_path ~/data --video_path kinetics_videos/jpg --annotation_path kinetics.json \ --result_path results --dataset kinetics --model resnet \ --model_depth 50 --n_classes 700 --batch_size 128 --n_threads 4 --checkpoint 5

关键参数说明

--model_depth：指定ResNet的深度（18, 34, 50, 101, 152, 200）
--n_classes：类别数量（Kinetics-700为700类）
--batch_size：批处理大小，根据GPU内存调整
--checkpoint：每隔多少个epoch保存一次模型

断点续训

如果训练中断，可以从上次保存的 checkpoint 继续：

python main.py --root_path ~/data --video_path kinetics_videos/jpg --annotation_path kinetics.json \ --result_path results --dataset kinetics --resume_path results/save_100.pth \ --model_depth 50 --n_classes 700 --batch_size 128 --n_threads 4 --checkpoint 5

🔍 模型推理与评估

训练完成后，我们可以使用训练好的模型进行动作识别推理，并评估模型性能。

执行推理

生成视频的Top-5类别概率：

python main.py --root_path ~/data --video_path kinetics_videos/jpg --annotation_path kinetics.json \ --result_path results --dataset kinetics --resume_path results/save_200.pth \ --model_depth 50 --n_classes 700 --n_threads 4 --no_train --no_val --inference --output_topk 5 --inference_batch_size 1

评估准确率

计算模型的Top-1准确率：

python -m util_scripts.eval_accuracy ~/data/kinetics.json ~/data/results/val.json --subset val -k 1 --ignore

📦 使用预训练模型进行迁移学习

项目提供了多个在大型数据集上预训练的模型，可以直接用于迁移学习，加速在自定义数据集上的训练。

预训练模型列表

部分可用的预训练模型包括：

r3d50_K_200ep.pth：在Kinetics-700上训练的ResNet-50
r3d50_KM_200ep.pth：在Kinetics-700+Moments in Time上训练的ResNet-50
r3d101_K_200ep.pth：在Kinetics-700上训练的ResNet-101

微调示例：UCF-101数据集

在UCF-101数据集上微调预训练模型：

python main.py --root_path ~/data --video_path ucf101_videos/jpg --annotation_path ucf101_01.json \ --result_path results --dataset ucf101 --n_classes 101 --n_pretrain_classes 700 \ --pretrain_path models/resnet-50-kinetics.pth --ft_begin_module fc \ --model resnet --model_depth 50 --batch_size 128 --n_threads 4 --checkpoint 5