【免费下载链接】PATH-ViTB
PATH-ViTB 是 OpenGVLab 开源的一个行人检测(pedestrian detection)ViT 预训练检查点仓库,基于 Hugging Face 模型仓库规范发布。仓库直接提供两份开箱即用的.pth检查点文件(ViT-Base 与 ViT-Large),文件名即训练配置,方便你在行人检测任务中快速复现与微调。
📦 PATH-ViTB 是什么?
PATH-ViTB 是一个纯检查点(Checkpoints)仓库:不包含训练代码,只托管训练好的模型权重。它的特点非常鲜明:
- ✅任务明确:文件名中的
peddet表示检查点用于行人检测(pedestrian detection)任务; - ✅双规格可选:同时提供
vitbase(轻量)和vitlarge(大容量)两种 ViT 骨干网络; - ✅配置透明:学习率、优化器、层衰减等超参数全部编码在文件名中,选型一目了然;
- ✅开源协议:根据 README.md 中的元数据,仓库采用OpenRAIL许可,并标注了
image-segmentation的流水线标签。
🗂 仓库结构与核心文件
仓库结构极其简洁,只有 3 个文件:
| 文件 | 说明 | 实际大小 |
|---|---|---|
| README.md | 仓库元数据(license、pipeline_tag) | < 1 KB |
| v100_32g_vitbase_...SharePosEmbed.pth | ViT-Base行人检测检查点,基于 CityPersons 数据集训练 | ≈ 327 MB |
| vitlarge_...SharePosEmbed.pth | ViT-Large行人检测检查点(DPR02 配置) | ≈ 1.13 GB |
💡 两份
.pth文件通过Git LFS托管,实际体积远大于指针文件,克隆时需要拉取 LFS 对象(见下文)。
🔍 如何读懂文件名里的训练配置
PATH-ViTB 的文件名本身就是一份“训练说明书”。下面以 ViT-Base 为例,拆解常用字段:
| 字段 | 含义 | 对应值 |
|---|---|---|
vitbase/vitlarge | ViT 骨干规格 | Base / Large |
size224 | 输入分辨率 | 224×224 |
lr1e3/lr5e4 | 初始学习率 | 1e-3 / 5e-4 |
stepLRx3 | 分步衰减学习率调度 | 3 次 step 衰减 |
adafactor | 优化器 | Adafactor |
clip05 | 梯度裁剪阈值 | 0.5 |
layerdecay075/layerdecay08 | 层学习率衰减系数 | 0.75 / 0.8 |
lpe | 可学习位置编码(Learned PE) | — |
LSA、reduct8、tbn1、heads2、gate1 | 注意力/解码器结构参数 | 实验配置 |
peddet_citypersons | 行人检测 + CityPersons 数据集 | 仅 Base 文件名体现 |
peddetShareDecoder | 行人检测共享解码器 | — |
peddetDPR02 | 行人检测扩展位置表示(DPR=0.2) | 仅 Large 文件名体现 |
SharePosEmbed | 共享位置嵌入实验设定 | — |
简单总结:Base 版本面向 CityPersons 行人检测任务,Large 版本采用 DPR02 位置表示策略,两者都是exp3实验设定下的产物。选型建议——显存有限选 Base,追求精度上限选 Large。
🚀 一键下载:如何拉取完整检查点
由于权重文件走 Git LFS,直接clone只会得到指针文件。正确姿势如下:
# 安装/初始化 Git LFS git lfs install # 克隆仓库并自动拉取 LFS 权重 git clone https://gitcode.com/OpenGVLab/PATH-ViTB如果之前已经克隆过仓库,只需补拉一次即可:
cd PATH-ViTB git lfs pull验证是否成功:用ls -lh查看,两个.pth文件应分别约为327 MB和1.13 GB(若只有几百字节,说明仍是 LFS 指针)。
⚠️ 使用注意事项
- 仓库只含权重,不含代码:模型结构定义、前向推理逻辑需配合你项目中的 ViT 实现来加载,加载时结构与文件名中的结构参数(如
heads2、reduct8)保持一致; - 协议合规:仓库采用OpenRAIL许可(见 README.md),商用前建议通读许可条款;
- 磁盘预算:两份检查点合计约1.4 GB,确保磁盘空间充足;
- 输入分辨率:两份检查点均以
size224训练,推理时建议保持 224×224 输入以匹配训练分布。
📌 总结
| 维度 | 结论 |
|---|---|
| 项目定位 | OpenGVLab 开源的行人检测 ViT 检查点仓库 |
| 核心资产 | ViT-Base(≈327 MB)与 ViT-Large(≈1.13 GB)两份.pth权重 |
| 适用任务 | 行人检测(peddet),Base 版基于 CityPersons 训练 |
| 获取方式 | git clone+git lfs pull |
| 开源协议 | OpenRAIL |
PATH-ViTB 用最精简的结构交付了最有价值的资产——两份“配置自描述”的高质量 ViT 检查点。对于正在做行人检测的开发者来说,它是快速搭建基线模型的理想起点 🎯
【免费下载链接】PATH-ViTB
相关推荐
DrRacket与Pie语言集成教程:启用智能提示与TODO列表管理
DrRacket与Pie语言集成教程:启用智能提示与TODO列表管理 欢迎来到这个完整的Pie语言开发环境配置指南!Pie是一个伴随《The Little Ty
ml5.js FaceApi 人脸检测与面部关键点识别 API 完全指南
ml5.js FaceApi 人脸检测与面部关键点识别 API 完全指南 ml5.js 的 faceApi 是一个面向浏览器的友好人脸检测接口,它封装了 fac
人工智能机器学习深度学习计算机视觉NLP人脸关键点检测:face-alignment库完整使用指南
人脸关键点检测:face alignment库完整使用指南 人脸关键点检测是计算机视觉中的重要技术,能够精确定位面部特征点如眼睛、鼻子、嘴巴等关键部位。face
人工智能计算机视觉深度学习
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考