nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

format

Browse source
This commit is contained in:
Shen-Chenhui 2024-06-06 07:45:35 +00:00
parent 53b9ccba1d
commit 918db183a2
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
eval/vbench_i2v/calc_vbench_i2v.py
View file

@ -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

2
eval/vbench_i2v/tabulate_vbench_i2v_scores.py
View file

@ -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,