jolibrain / joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications
https://www.joligen.com
Other
243 stars 32 forks source link

add option for vid inference #712

Closed wr0124 closed 1 week ago

wr0124 commented 1 week ago

Add an option --vid_fps to specify the frames per second (FPS) for video generation, and an option --vid_frame_extension_number to include an additional frame for inference. it works with

cd scripts/
mkdir -p ../inference_mario/1017   && \
python3 -W ignore::UserWarning  gen_vid_diffusion.py  \
--img_in   /data1/paths_part.txt  \
--mask_in /paths_part.txt    \
--model_in_file  path/to/latest_net_G_A.pth  \
--paths_in_file   path/to/paths.txt\
--dir_out  ../inference_mario/1017  \
--sampling_steps 3  \
--gpuid 0  \
--vid_fps 1  \
--vid_frame_extension_number  6 \