diff --git a/eval/vbench_i2v/calc_vbench_i2v.py b/eval/vbench_i2v/calc_vbench_i2v.py index fbbdd11..5692fac 100644 --- a/eval/vbench_i2v/calc_vbench_i2v.py +++ b/eval/vbench_i2v/calc_vbench_i2v.py @@ -1,6 +1,8 @@ import argparse import os +import torch + from vbench import VBench from vbench2_beta_i2v import VBenchI2V @@ -31,14 +33,14 @@ if __name__ == "__main__": os.makedirs(output_dir, exist_ok=True) video_path = args.video_folder - my_VBench_I2V = VBenchI2V("cuda", full_info_path, output_dir) + my_VBench_I2V = VBenchI2V(torch.device("cuda"), full_info_path, output_dir) for i2v_dim in i2v_dimensions: my_VBench_I2V.evaluate(videos_path=video_path, name=i2v_dim, dimension_list=[i2v_dim], resolution="1-1") kwargs = {} kwargs["imaging_quality_preprocessing_mode"] = "longer" # use VBench/evaluate.py default - my_VBench = VBench("cuda", full_info_path, output_dir) + my_VBench = VBench(torch.device("cuda"), full_info_path, output_dir) for quality_dim in video_quality_dimensions: my_VBench.evaluate( videos_path=video_path, name=quality_dim, dimension_list=[quality_dim], mode="vbench_standard", **kwargs diff --git a/eval/vbench_i2v/tabulate_vbench_i2v_scores.py b/eval/vbench_i2v/tabulate_vbench_i2v_scores.py index fb6a518..8a9fdc1 100644 --- a/eval/vbench_i2v/tabulate_vbench_i2v_scores.py +++ b/eval/vbench_i2v/tabulate_vbench_i2v_scores.py @@ -23,7 +23,7 @@ I2V_QUALITY_LIST = [ DIM_WEIGHT_I2V = { "camera_motion": 0.1, "i2v_subject": 1, - "i2v_background_consistency": 1, + "i2v_background": 1, "subject_consistency": 1, "background_consistency": 1, "motion_smoothness": 1,