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 setobj_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 testMAX_KPT_NUM: int, maximum kpt_num in an image