JinyuLiu/Video-DETR-demo

Video-DETR Demo

上传一段视频和一句文本查询,定位视频中最相关的片段,并给出每个片段的高光分数。不输入查询时,默认做通用高光检测。

安装依赖

venv/(Python 依赖)因文件数量过多未包含在仓库中，需要自己创建；其余环境——ffmpeg-tools/ 中的 ffmpeg、模型权重、示例视频——都已内置，无需处理。

cd Video-DETR-demo
python3.11 -m venv venv
venv/bin/pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

# GPU (CUDA 12.4):
venv/bin/pip install torch==2.5.1+cu124 torchvision==0.20.1+cu124 --index-url https://mirrors.tuna.tsinghua.edu.cn/pytorch-wheels/cu124

# 没有 GPU:
# venv/bin/pip install torch==2.5.1 torchvision==0.20.1 --index-url https://mirrors.tuna.tsinghua.edu.cn/pytorch-wheels/cpu

使用方法

./run_webapp.sh

启动后访问 http://localhost:7860 即可。脚本会自动清理上一次残留的进程,并在端口被占用时给出提示。没有 GPU 也可以运行，自动回退到 CPU。

默认使用 CUDA_VISIBLE_DEVICES=0,如需更换 GPU:

CUDA_VISIBLE_DEVICES=1 ./run_webapp.sh