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 \