reid.datasets

reid.datasets.create(name, root, *args, **kwargs)[source]

Create a dataset instance.

Parameters:
  • name (str) – The dataset name. Can be one of ‘viper’, ‘cuhk01’, ‘cuhk03’, ‘market1501’, and ‘dukemtmc’.
  • root (str) – The path to the dataset directory.
  • split_id (int, optional) – The index of data split. Default: 0
  • num_val (int or float, optional) – When int, it means the number of validation identities. When float, it means the proportion of validation to all the trainval. Default: 100
  • download (bool, optional) – If True, will download the dataset. Default: False
class reid.datasets.CUHK01(root, split_id=0, num_val=100, download=True)[source]
class reid.datasets.CUHK03(root, split_id=0, num_val=100, download=True)[source]
class reid.datasets.DukeMTMC(root, split_id=0, num_val=100, download=True)[source]
class reid.datasets.Market1501(root, split_id=0, num_val=100, download=True)[source]
class reid.datasets.VIPeR(root, split_id=0, num_val=100, download=True)[source]