nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-07 04:41:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

eval

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-30 14:21:46 +08:00
parent e827b78000
commit d4e03478b4
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
eval/eval_common_metric.py
View file

@ -42,11 +42,11 @@ from torchvision.transforms import Lambda, Compose
from torchvision.transforms._transforms_video import CenterCropVideo
import sys
sys.path.append(".")
from opensora.eval.cal_lpips import calculate_lpips
from opensora.eval.cal_fvd import calculate_fvd
from opensora.eval.cal_psnr import calculate_psnr
from opensora.eval.cal_flolpips import calculate_flolpips
from opensora.eval.cal_ssim import calculate_ssim
from .cal_lpips import calculate_lpips
from .cal_fvd import calculate_fvd
from .cal_psnr import calculate_psnr
from .cal_flolpips import calculate_flolpips
from .cal_ssim import calculate_ssim
try:
from tqdm import tqdm

2
eval/script/eval.sh
View file

@ -1,6 +1,6 @@
python eval/eval_common_metric.py \
--batch_size 2 \
--real_video_dir ..//test_eval/release/origin \
--real_video_dir ../test_eval/release/origin \
--generated_video_dir ../test_eval/release \
--device cuda \
--sample_fps 10 \