From 918db183a2ee0d8042f70a8832fcfe54ff9056af Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Thu, 6 Jun 2024 07:45:35 +0000 Subject: [PATCH] format --- eval/vbench_i2v/calc_vbench_i2v.py | 6 ++++-- eval/vbench_i2v/tabulate_vbench_i2v_scores.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) 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,