PascalVOC

class pygmtools.dataset.PascalVOC(sets, obj_resize, **ds_dict)

Download and preprocess PascalVOC Keypoint 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 5 params(keys) for PascalVOC:

  • KPT_ANNO_DIR: str, directory of keypoint annotations

  • ROOT_DIR: str, directory of data

  • SET_SPLIT: str, set split path

  • CLASSES: list, data classes

  • CACHE_PATH: str, directory of data cache

download(url=None, name=None, retries=5)

Automatically download PascalVOC dataset.

Parameters

• url – str, web url of PascalVOC and PascalVOC annotation

• name – str, "PascalVOC" to download PascalVOC and "PascalVOC_anno" to download PascalVOC annotation

process()

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