TPO dataset for LongVA
收藏资源简介:
TPO数据集用于LongVA模型,包含10k条数据,用于训练视频大型多模态模型(video-LMMs),以增强视频理解能力。
The TPO dataset is utilized for the LongVA model, comprising 10k data entries, which are employed to train video large-scale multimodal models (video-LMMs) to enhance video comprehension capabilities.
Temporal Preference Optimization (TPO) 数据集概述
数据集基本信息
- 名称: Temporal Preference Optimization (TPO) for Long-Form Video Understanding
- 用途: 用于长视频理解的自训练时序偏好优化
- 数据集地址: Huggingface Dataset
数据集内容
- 数据生成方法:
- 局部TPO: 生成针对短视频段的查询,对比响应保留或排除目标片段。
- 综合TPO: 生成针对更广泛理解的查询,使用完整视频与稀疏降采样视频进行对比响应。
- 数据用途: 作为偏好数据集训练视频大型多模态模型(video-LMMs),引导模型优先选择偏好响应以提升视频理解能力。
模型权重
安装与使用
-
LongVA-TPO安装: bash git clone https://github.com/ruili33/TPO cd TPO conda create -n TPOLongVA python=3.10 conda activate TPOLongVA pip install torch==2.1.2 torchvision --index-url https://download.pytorch.org/whl/cu118 pip install -e "longva/.[train]" pip install packaging && pip install ninja && pip install flash-attn==2.5.0 --no-build-isolation --no-cache-dir pip install -r requirements_longva.txt
-
LLaVA-Video-TPO安装: bash conda create -n TPOllava python=3.10 -y conda activate TPOllava pip install --upgrade pip pip install -e "LLaVA/.[train]" pip install flash-attn==2.5.0 --no-build-isolation --no-cache-dir
推理与评估
- 推理:
- LongVA-TPO: 参考
longva/inference_longva.py - LLaVA-Video-TPO: 参考
LLaVA/inference_llava.py
- LongVA-TPO: 参考
- 评估:
- 使用 lmms-eval 进行评估
- LongVA-TPO: 参考
longva/eval.sh - LLaVA-Video-TPO: 参考
LLaVA/eval.sh
待发布内容
- 时序偏好数据整理流程(预计3月发布)
- 训练代码(预计3月发布)




