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

feat(ml): inference_vid with canny #696

Closed wr0124 closed 1 month ago

wr0124 commented 1 month ago

Video Inference Debugging 1 With Canny Edge Detection:

2 Without Canny Edge Detection:

cd scripts/ mkdir -p ../inference_mario_bz python3 -W ignore::UserWarning gen_vid_diffusion.py \ --model_in_file /path/to/checkpoints/test_mario_vid_example_canny_drop2/latest_net_G_A.pth \ --img_in /path/to/paths_part.txt \ --paths_in_file /path/to/online_mario2sonic_full_mario/trainA/paths.txt \ --mask_in /paths_part.txt \ --dir_out ../inference_mario_bz \ --sampling_steps 1000 \ --gpuid 2 \ ~