RaivoKoot / Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

demo fix error

nchamroukhi opened this issue · comments

in line 131 just replace frames with frame_tensor:
plot_video(rows=1, cols=5, frame_list= frame_tensor, plot_width=15., plot_height=3.)>

Thank you! Fixed!