activitynet-stage2 / README.md
happy8825's picture
Add video-text-to-text pipeline tag, link paper, project page, and code to model card (#1)
918ca41
|
Raw
History Blame Contribute Delete
1.41 kB
metadata
base_model: Qwen/Qwen3-VL-4B-Instruct
library_name: transformers
license: apache-2.0
pipeline_tag: video-text-to-text
tags:
  - video-retrieval
  - temporal-grounding
  - videosearch-r1

VideoSearch-R1 ActivityNet Stage 2

This is the Stage 2 VideoSearch-R1 checkpoint trained for ActivityNet, presented in the paper VideoSearch-R1: Iterative Video Retrieval and Reasoning via Soft Query Refinement.

Stage 2 starts from the ActivityNet Stage 1 checkpoint and optimizes iterative retrieval and temporal grounding behavior with the VideoSearch-R1 training pipeline.

Usage

Use with the VideoSearch-R1 codebase:

bash scripts/data_construct/download_preextracted_data.bash activitynet
EVAL_GPUS=0 bash scripts/inference/inference.bash activitynet --checkpoint VideoSearchR1/activitynet-stage2

Citation

@inproceedings{lee2026videosearchr1,
  title     = {VideoSearch-R1: Iterative Video Retrieval and Reasoning via Soft Query Refinement},
  author    = {Lee, Seohyun and Choi, Seoung and Ko, Dohwan and Kim, Jongha and Kim, Hyunwoo J.},
  booktitle = {European Conference on Computer Vision (ECCV)},
  year      = {2026}
}