nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-12 13:54:53 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

update eval readme

Browse source
This commit is contained in:
Shen-Chenhui 2024-05-03 08:20:26 +00:00
parent a2ef63922f
commit 9e583cbf2d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
eval/README.md
View file

@ -56,6 +56,6 @@ pip install scipy
### Commands: carefule to change the setting to training setting
```bash
# metric can either be: ssim, psnr, lpips, flolpips
python eval/vae/eval_common_metric.py --batch_size 2 --real_video_dir <path/to/original/videos> --generated_video_dir <path/to/generated/videos> --device cuda --sample_fps 24 --crop_size 256 --resolution 256 --num_frames 17 --sample_rate 1 --metric <metric>
# metric can any one or list of: ssim, psnr, lpips, flolpips
python eval/vae/eval_common_metric.py --batch_size 2 --real_video_dir <path/to/original/videos> --generated_video_dir <path/to/generated/videos> --device cuda --sample_fps 24 --crop_size 256 --resolution 256 --num_frames 17 --sample_rate 1 --metric ssim psnr lpips flolpips
```