ECVL 0.4.0a
Pre-release
Pre-release
MicheleCancilla
released this
09 Jun 14:21
·
58 commits
to master
since this release
This pre-release implement a parallel batch loader that instantiates several producers (workers) and allows to load images while running the training code. The loading + augmentation function used by the producers, being virtual, can be easily reimplemented to modify its behavior as desired.
Main changes:
- Splits can now have names different from training, validation, test and they can count more than three.
- DatasetAugmentations must be as long as the number of splits that are in the YAML file.
- In the DLDataset constructor the user can specify whether to automatically drop the last samples of the split if they are not enough to compose a batch, set the number of workers and the ratio that will be used to calculate the maximum dimension of the queue.
- ResetCurrentBatch became ResetBatch. It can be specified for which split you want to reset the batch counter and if you want to shuffle the batch.
- ColorType for loaded images in DLDataset is now RGB by default (instead of BGR).