From d4e03478b4e69c69b3fb056a0864917d8708b36a Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Tue, 30 Apr 2024 14:21:46 +0800 Subject: [PATCH] eval --- eval/eval_common_metric.py | 10 +++++----- eval/script/eval.sh | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/eval/eval_common_metric.py b/eval/eval_common_metric.py index 452c032..abca5fa 100644 --- a/eval/eval_common_metric.py +++ b/eval/eval_common_metric.py @@ -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 diff --git a/eval/script/eval.sh b/eval/script/eval.sh index d904699..c1d1dbc 100644 --- a/eval/script/eval.sh +++ b/eval/script/eval.sh @@ -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 \