TrainSet

class lib.training.data.data_set.TrainSet(side: str, image_folder: str, size: int)

Bases: _BaseSet

Base class for Training and Preview dataset loaders to inherit from

Parameters:

• side (str) – The side of the model (“A”, “B” etc.)

• image_folder (str) – Full path to a folder containing training images

• size (int) – The size to return samples at. This should be the maximum of the model input/output size for train sets or the model input size for preview sets