SPair71k¶
- class pygmtools.dataset.SPair71k(sets, obj_resize, problem='2GM', ds_dict=None, **args)[source]¶
This class is defined to download and preprocess SPair71k dataset.
- Parameters
sets – str, problem set, ‘train’ for training set and ‘test’ for test set
obj_resize – tuple, resized image size
problem – str, problem type, only ‘2GM’ in SPair71k
ds_dict –
settings of dataset, containing at most 4 params(keys) for SPair71k:
TRAIN_DIFF_PARAMS: list of images that should be dumped in train set
EVAL_DIFF_PARAMS: list of images that should be dumped in test set
COMB_CLS: bool, whether to combine images in different classes
SIZE: str, ‘large’ for SPair71k-large and ‘small’ for SPair71k-small