IMC_PT_SparseGM

class pygmtools.dataset.IMC_PT_SparseGM(sets, obj_resize, **ds_dict)[source]

Download and preprocess IMC_PT_SparseGM dataset.

Parameters
  • sets – str, problem set, 'train' for training set and 'test' for testing set

  • obj_resize – tuple, resized image size

  • ds_dict

    settings of dataset, containing at most 4 params(keys) for IMC_PT_SparseGM:

    • ROOT_DIR_IMG: str, directory of images

    • ROOT_DIR_NPZ: str, directory of annotations

    • CLASSES: dict, classes of training and test data, keys: 'train' for training and 'test' for test

    • MAX_KPT_NUM: int, maximum kpt_num in an image

download(url=None, retries=15)[source]

Automatically download IMC_PT_SparseGM dataset.

Parameters

url – str, web url of IMC_PT_SparseGM

process()[source]

Process the dataset and generate data-(size, size).json for preprocessed dataset, train.json for training set, and test.json for testing set.