compressai.datasets#
ImageFolder#
- class compressai.datasets.ImageFolder(root, transform=None, split='train')[source]#
Load an image folder database. Training and testing image samples are respectively stored in separate directories:
- rootdir/ - train/ - img000.png - img001.png - test/ - img000.png - img001.png
- Parameters:
root (string) – root directory of the dataset
transform (callable, optional) – a function or transform that takes in a PIL image and returns a transformed version
split (string) – split mode (‘train’ or ‘val’)
VideoFolder#
- class compressai.datasets.VideoFolder(root, rnd_interval=False, rnd_temp_order=False, transform=None, split='train')[source]#
Load a video folder database. Training and testing video clips are stored in a directorie containing mnay sub-directorie like Vimeo90K Dataset:
- rootdir/ train.list test.list - sequences/ - 00010/ ... -0932/ -0933/ ... - 00011/ ... - 00012/ ...
training and testing (valid) clips are withdrew from sub-directory navigated by corresponding input files listing relevant folders.
This class returns a set of three video frames in a tuple. Random interval can be applied to if subfolders includes more than 6 frames.
- Parameters:
root (string) – root directory of the dataset
rnd_interval (bool) – enable random interval [1,2,3] when drawing sample frames
transform (callable, optional) – a function or transform that takes in a PIL image and returns a transformed version
split (string) – split mode (‘train’ or ‘test’)