result rendering

[farhad-abdi]

Hi, my issue was not really significant to open a thread. I just want to know how to display final sampled_videos using matplotlib, i'm permuting (channels, frames, height, width) using imgs = sampled_videos[0].permute(1,2,3,0) but it's showing noise only!

[SamuelSchmidgall]

You can try the following:

import imageio
import numpy as np

sampled_videos = diffusion.sample(batch_size=1)
sampled_videos = sampled_videos.squeeze(0).permute(1, 2, 3, 0).detach().cpu().numpy()
sampled_videos = ((sampled_videos - sampled_videos.min()) / (sampled_videos.max() - sampled_videos.min()) * 255).astype(np.uint8)
imageio.mimsave('output_video.gif', sampled_videos, fps=5)