☰
如何快速上手InternVideo2-Stage1-1B-224p-f8-SthSth:从权重下载到视频推理的5步完整教程
2026/10/17 10:31:58 网站建设 项目流程

【免费下载链接】InternVideo2-Stage1-1B-224p-f8-SthSth

项目地址:https://ai.gitcode.com/OpenGVLab/InternVideo2-Stage1-1B-224p-f8-SthSth
点击查看免费下载

🎬 InternVideo2-Stage1-1B-224p-f8-SthSth 是 InternVideo2 视频基础模型的官方 Stage1 微调权重(1B 参数规模,224p 输入分辨率,f8 表示时空采样 8 帧,SthSth 即 Spatio-Temporal 时空理解任务)。本仓库只包含 PyTorch 权重文件,本文将用 5 个步骤带你从零完成:环境准备 → 克隆权重 → 校验完整性 → 加载配置 → 跑通视频推理,全程约 10 分钟。

第1步:检查Python环境与GPU依赖

在开始之前,先确认本机环境满足以下要求:

依赖最低要求说明
Python3.8+推荐使用 3.10
PyTorch2.0+configuration.json 声明framework: pytorch
GPU 显存8 GB+1B 参数模型推理推荐 8GB 起步

用下面两条命令快速自检:

python -V python -c "import torch; print(torch.__version__, torch.cuda.is_available())"

💡 如果输出False(无 CUDA),模型可以运行在 CPU 上,但推理速度会明显变慢。

第2步:克隆InternVideo2-Stage1权重仓库

执行以下命令把权重仓库拉到本地:

git clone https://gitcode.com/OpenGVLab/InternVideo2-Stage1-1B-224p-f8-SthSth

克隆完成后,目录下应该有这 4 个文件:

  • 1B_ft_ssv1_f8.pth —— SSV1 视频编码器分支的 Stage1 微调权重
  • 1B_ft_ssv2_f8.pth —— SSV2 视频编码器分支的 Stage1 微调权重
  • README.md —— 模型卡,介绍项目来源与论文信息
  • configuration.json —— 仓库元配置

第3步:校验权重文件完整性(新手必做)

⚠️ 这是最容易踩的坑:这两个.pth文件通过 Git LFS 分发,真实大小应约为 1.9 GB(约 2,041,791,213 字节)。如果 clone 时未安装git-lfs,本地拿到的只是几十字节的 LFS 指针文本,模型将无法加载。

用一条命令检查文件大小:

ls -lh 1B_ft_ssv1_f8.pth 1B_ft_ssv2_f8.pth

正确结果应显示1.9G左右。若只有几百字节,先安装 LFS 再拉取真实文件:

git lfs install git lfs pull

你也可以对照文件内的 sha256 校验值(如c4f3591e...)用sha256sum二次验证,确保权重未被截断或损坏。

第4步:加载权重与仓库配置

这个仓库是"纯权重仓库",模型结构代码位于上游 InternVideo2 项目(参见 README.md 中的模型来源说明)。典型加载方式是配合官方 InternVideo2 代码库,用torch.load读取权重:

import torch # 加载 Stage1 微调权重(以 ssv1 分支为例) state_dict = torch.load( "1B_ft_ssv1_f8.pth", map_location="cpu" ) # 随后加载到 InternVideo2 模型对应模块即可

关于仓库配置,configuration.json 内容如下,说明该权重面向 PyTorch 框架、任务类型为others(视频理解特征/表征类任务):

{"framework": "pytorch", "task": "others", "allow_remote": true}

命名速查,帮你理解每个参数的含义:

  • 1B:约 10 亿参数
  • ft:fine-tuned,Stage1 阶段微调产物
  • ssv1 / ssv2:InternVideo2 的两路视频编码器分支
  • 224p:输入视频帧缩放到 224×224
  • f8:按时间维采样 8 帧送入模型
  • SthSth:Spatio-Temporal,时空定位类理解任务

第5步:跑通视频推理

拿到权重与结构代码后,推理流程非常直接:

  1. 准备视频:任选一段本地视频(mp4 均可),InternVideo2 会对视频按 f8 规则采样 8 帧、缩放到 224p 输入;
  2. 前向推理:用官方 InternVideo2 推理脚本加载上述.pth权重,输入视频得到时空理解特征/定位结果;
  3. 按需选择分支:ssv1 与 ssv2 两套权重对应不同编码器分支,可按任务需求各测一遍做对比。

🚀 小贴士:首次推理包含权重加载与编译缓存,耐心等待 1~2 分钟属正常现象;长视频建议先降低采样密度再逐步调高。

常见问题排查清单

现象原因解决办法
权重文件只有几百字节未安装 git-lfs,拉到的是 LFS 指针执行git lfs install && git lfs pull
加载时显存不足GPU 显存 < 8GB使用map_location="cpu"或换更大显存设备
加载报 missing/unexpected keys权重与模型结构版本不匹配确认使用 Stage1 对应的模型结构代码

总结:5步上手回顾

1️⃣ 环境自检(Python + PyTorch + CUDA) 2️⃣ 克隆权重仓库 3️⃣ 用ls -lh与 sha256 校验 1.9GB 权重完整性 4️⃣torch.load加载 1B_ft_ssv1_f8.pth / 1B_ft_ssv2_f8.pth 并核对 configuration.json 5️⃣ 配合官方代码库完成视频时空推理

到这里,你就完成了从权重下载到视频推理的完整闭环。建议先跑通 ssv1 分支,再横向对比 ssv2 的效果,感受 InternVideo2 双分支设计在时空理解任务上的差异。祝实验顺利!🏁

【免费下载链接】InternVideo2-Stage1-1B-224p-f8-SthSth

项目地址:https://ai.gitcode.com/OpenGVLab/InternVideo2-Stage1-1B-224p-f8-SthSth
点击查看免费下载
上一篇:PyPTO CODEGEN 组件经验:3 层嵌套 pypto.loop 的 workspace 累积失效与 parallel=True 修复实战
下一篇:PyPTO 算子调优实战:leafHash 到前端代码映射方法与 sg_set_scope 合图插入点定位

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询