vbench-i2v / run.sh
lxl-158's picture
upload vbench-i2v
d0d70d4 verified
#!/bin/bash
#
# VBench I2V 完整流程: 采样 -> 评测 -> 计算分数
#
# 用法:
# bash run.sh <model_name> [resolution]
#
# 示例:
# bash run.sh my_model
# bash run.sh my_model 16-9
set -e
MODEL_NAME=${1:?请提供模型名称}
RESOLUTION=${2:-16-9}
echo "模型: $MODEL_NAME"
echo "分辨率: $RESOLUTION"
echo ""
# 激活环境
source vbench_env/bin/activate
# 1. 采样
echo "========== [1/3] 采样 =========="
python run_sample.py --model_name "$MODEL_NAME" --resolution "$RESOLUTION"
# 2. 评测
echo ""
echo "========== [2/3] 评测 =========="
python run_eval.py --model_name "$MODEL_NAME" --resolution "$RESOLUTION"
# 3. 计算分数
echo ""
echo "========== [3/3] 计算分数 =========="
python cal_score.py --model_name "$MODEL_NAME"
echo ""
echo "========== 完成 =========="
echo "视频: ./sampled_videos/$MODEL_NAME/"
echo "评测: ./evaluation_results/$MODEL_NAME/"