search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.84k stars 178 forks source link

Load sequences of images #191

Open vincent-leguen opened 2 years ago

vincent-leguen commented 2 years ago

Thank you for this great work ! I'm wondering if it can be used to efficiently load sequences of images, for example for video prediction. Imagine you have a large dataset of images from cameras (ordered images, eg. one folder per day) and you want to sample a batch of B samples, each sample being a sequence of p+f consecutive images (p input frames, f future frames to predict). Is there an efficient way to do this ?

richardrl commented 5 months ago

@GuillaumeLeclerc Hey was wondering if there's any update on this?

volcverse commented 4 months ago

Hello, is there any tutorials about this? I check the doc but can't find anything related.

richardrl commented 3 months ago

@volcverse You just need to write a custom TraversalOrder that associates sequences of images together in each minibatch. #374 allows feeding any custom data you need to do this into your custom TraversalOrder class