Vjepa_mamba_dataset_v2
收藏数据集概述
- 数据集名称: V-JEPA Mamba Dataset (v2)
- 许可协议: MIT
- 任务类别: 视频分类、图像特征提取、其他
- 标签: 自我中心、视频、帧、Mamba、V-JEPA、预训练
- 数据集大小: 100K < n < 1M(样本数量)
数据集来源与预处理
该数据集来源于 Egocentric-10K 数据集,对每个视频进行以下过滤和裁剪:
- 时长过滤: 丢弃时长小于3分钟的视频;超过3分钟的视频截取前3分钟。
- 帧提取: 使用 ffmpeg 以 8 fps 提取帧,每个视频生成1,440帧。
- 分辨率: 384×384 RGB,采用最短边缩放加中心裁剪。
- 存储格式: 每个视频存储为一个单独的 Parquet 文件。
数据字段
每个 Parquet 文件包含以下列:
| 列名 | 类型 | 描述 |
|---|---|---|
video_index |
int32 | 视频唯一索引(从0开始) |
frame_index |
int16 | 帧在视频中的位置(0-1439) |
frame_bytes |
binary | JPEG 编码的帧(质量92) |
video_key |
string | 来自 Egocentric-10K 的原始视频标识符 |
数据集结构
data/ video_00000.parquet # 视频 0-9999 ... _state.json # 恢复检查点 shard_01/ video_10000.parquet # 视频 10000-19999 ...
(分片存储在目录中,每个目录约9,500个文件,以遵守 Hugging Face 的10k/目录限制)
使用示例
python import pyarrow.parquet as pq from PIL import Image import io
加载视频的帧
table = pq.read_table("data/shard_01/video_10238.parquet") for row in table.to_pylist(): img = Image.open(io.BytesIO(row["frame_bytes"])) # 384×384 RGB # ... 训练循环 ...
处理流程
预处理使用 preprocess-v3.py 脚本实现:
- 下载与解压: 8个并行下载线程 + 8个解压线程,通过队列解耦。
- 处理: 使用所有可用的 CPU 核心(30个保留),包括 ffmpeg 解码 → JPEG 编码 → 存储为 Parquet 文件。
- 上传: 通过
upload_folder每个块单独提交(避免速率限制)。
恢复安全机制:_state.json 跟踪 tar 级别的进度。重启时,已完全处理的 tar 文件被跳过(零重新下载)。
来源与引用
- 源数据集: builddotai/Egocentric-10K — 包含85个工厂的19,495个 tar 分片,记录自我中心视频。
- 引用格式:
bibtex @misc{vjepa_mamba_dataset, title={V-JEPA Mamba Dataset: Preprocessed Egocentric Video Frames}, author={Phi-9 Research}, year={2026}, url={https://huggingface.co/datasets/rookierufus/Vjepa_mamba_dataset_v2} }




