sdzt's picture
Add source code
33569f9 verified
|
Raw
History Blame Contribute Delete
5 kB
# Dataset Preparation
## Training data
- annotation: We place the annotation file in [dataset/trainval/train_2k5.json](../dataset/trainval/train_2k5.json).
- original videos: You can download download from our organized version in [Boshenxx/TimeR1-Dataset](https://huggingface.co/datasets/Boshenxx/TimeR1-Dataset);
or you can download and organize the original data from [VTG-IT](https://huggingface.co/datasets/Yongxin-Guo/VTG-IT), [TimeIT](https://huggingface.co/datasets/ShuhuaiRen/TimeIT), [TimePro](https://huggingface.co/Lanxingxuan/TimeSuite), [HTStep](https://openreview.net/pdf?id=vv3cocNsEK) and [LongVid](https://huggingface.co/datasets/OpenGVLab/LongVid).
## Testing data
folder structure:
```
dataset
β”œβ”€ timer1
β”‚ β”œβ”€ annotations
β”‚ β”‚ β”œβ”€ train_2k5.json
β”‚ β”‚ └─ tvgbench.json
β”‚ β”œβ”€ videos
β”‚ β”‚ β”œβ”€ timerft_data
β”‚ β”‚ | β”œβ”€ xxx.mp4
β”‚ β”‚ β”‚ └─ ...
β”‚ β”‚ β”œβ”€ tvgbench_data
β”‚ β”‚ | β”œβ”€ xxx.mp4
β”‚ β”‚ β”‚ └─ ...
β”œβ”€ activitynet
β”‚ β”œβ”€ annotations
β”‚ β”‚ β”œβ”€ sentence_temporal_grounding
β”‚ β”‚ β”‚ └─ test.json
β”‚ β”œβ”€ videos
β”‚ | β”œβ”€ v_zzz_3yWpTXo.mp4
β”‚ β”‚ └─ ...
β”œβ”€ charades
β”‚ β”œβ”€ Charades_anno
β”‚ β”‚ └─ Charades_v1_test.csv
β”‚ β”œβ”€ Charades_v1
β”‚ | β”œβ”€ 0I0FX.mp4
β”‚ β”‚ └─ ...
β”œβ”€ mvbench
β”‚ β”œβ”€ json
β”‚ β”‚ β”œβ”€ action_antonym.json
β”‚ β”‚ └─ ...
β”‚ β”œβ”€ videos
β”‚ β”‚ β”œβ”€ clevrer
β”‚ β”‚ └─ ...
β”œβ”€ tempcompass
β”‚ β”œβ”€ questions
β”‚ β”‚ β”œβ”€ multi-choice.json
β”‚ β”‚ └─ ...
β”‚ β”œβ”€ videos
β”‚ β”‚ β”œβ”€ 315784.mp4
β”‚ β”‚ └─ ...
β”œβ”€ egoschema
β”‚ β”œβ”€ MC
β”‚ β”‚ └─ test-00000-of-00001.parquet
β”‚ β”œβ”€ Subset
β”‚ β”‚ └─ test-00000-of-00001.parquet
β”‚ β”œβ”€ videos
β”‚ β”‚ β”œβ”€ 001934bb-81bd-4cd8-a574-0472ef3f6678.mp4
β”‚ β”‚ └─ ...
β”œβ”€ videomme
β”‚ β”œβ”€ videomme
β”‚ β”‚ └─ test-00000-of-00001.parquet
β”‚ β”œβ”€ data
β”‚ β”‚ β”œβ”€ _8lBR0E_Tx8.mp4
└─ └─ └─ ...
```
### ActivityNet
Download link: [ActivityNet](https://cs.stanford.edu/people/ranjaykrishna/densevid/)
For fine-tuning setting, before training, you need to preprocess the video data.
```bash
bash preprocess_video.sh
```
Specify the path to the Charades-STA dataset (video files, annotations, etc.).
### Charades
Download link: [Charades-v1](https://huggingface.co/datasets/HuggingFaceM4/charades)
For fine-tuning setting, before training, you need to preprocess the video data.
```bash
bash preprocess_video.sh
```
Specify the path to the Charades-STA dataset (video files, annotations, etc.).
### TVGBench
Download link: [hf: Boshenxx/TimeR1-Dataset](https://huggingface.co/datasets/Boshenxx/TimeR1-Dataset)
### MVBench
We place the annotation file of tvgbench in [dataset/trainval/tvgbench.json](../dataset/trainval/tvgbench.json).
Download link: [hf: OpenGVLab/MVBench](https://huggingface.co/datasets/OpenGVLab/MVBench)
### VideoMME
Download link: [hf: lmms-lab/MVBench](https://huggingface.co/datasets/lmms-lab/Video-MME)
### Egoschema
Download link: [hf: lmms-lab/egoschema](https://huggingface.co/datasets/lmms-lab/egoschema)
### TempCompass
Download link: [hf: lmms-lab/TempCompass](https://huggingface.co/datasets/lmms-lab/TempCompass)