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-05 09:09:53 +00:00
parent ac97bc6baa
commit 9dc63b35ca
4 changed files with 1121 additions and 1121 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2236
assets/texts/VBench/all_i2v.txt
View file

File diff suppressed because it is too large Load diff

2
docs/installation.md
View file

@ -74,7 +74,7 @@ pip install -v .[eval]
You need to manually install [VBench](https://github.com/Vchitect/VBench):
```bash
pip install --no-deps vbench==0.1.1
pip install --no-deps vbench==0.1.2
# If the installation shows a warning about the intalled vbench not in PATH, you need to add it by:
export PATH="/path/to/vbench:$PATH"
```

2
eval/vbench_i2v/json_to_txt.py
View file

@ -10,7 +10,7 @@ save_path = "all_i2v.txt"
data = json.load(open(json_file))
txt = [
f'{x["prompt_en"]}{{"reference_path": "{os.path.join(cache_root, resolution, x["image_name"])}", "mask_strategy": "0,0,0,1,0"}}'
f'{x["prompt_en"]}{{"reference_path": "{os.path.join(cache_root, resolution, x["image_name"])}", "mask_strategy": "0"}}'
for x in data
]
with open(save_path, "w") as f:

2
eval/vbench_i2v/vbench_i2v.py
View file

@ -29,7 +29,7 @@ if __name__ == "__main__":
args = parse_args()
output_dir = os.path.join(args.model_ckpt, "eval", "vbench_i2v")
os.makedirs(output_dir, exist_ok=True)
video_path = args.video_path
video_path = args.video_folder
my_VBench_I2V = VBenchI2V("cuda", FULL_INFO_PATH, "evaluation_results")
my_VBench_I2V.evaluate(videos_path=video_path, name="vbench_i2v", dimension_list=I2V_DIMENSIONS, resolution="1-1")