mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add global keyword

pull/201/head
KaiyangZhou 2019-05-24 17:13:30 +01:00
parent 5ae81344a0
commit 31068ca233
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
torchreid/data/datasets/__init__.py
View File

@ -32,7 +32,8 @@ def init_image_dataset(name, **kwargs):
"""Initializes an image dataset.""" """Initializes an image dataset."""
avai_datasets = list(__image_datasets.keys()) avai_datasets = list(__image_datasets.keys())
if name not in avai_datasets: if name not in avai_datasets:
raise ValueError('Invalid dataset name. Received "{}", but expected to be one of {}'.format(name, avai_datasets)) raise ValueError('Invalid dataset name. Received "{}", '
'but expected to be one of {}'.format(name, avai_datasets))
return __image_datasets[name](**kwargs) return __image_datasets[name](**kwargs)
@ -40,16 +41,17 @@ def init_video_dataset(name, **kwargs):
"""Initializes a video dataset.""" """Initializes a video dataset."""
avai_datasets = list(__video_datasets.keys()) avai_datasets = list(__video_datasets.keys())
if name not in avai_datasets: if name not in avai_datasets:
raise ValueError('Invalid dataset name. Received "{}", but expected to be one of {}'.format(name, avai_datasets)) raise ValueError('Invalid dataset name. Received "{}", '
'but expected to be one of {}'.format(name, avai_datasets))
return __video_datasets[name](**kwargs) return __video_datasets[name](**kwargs)
def register_image_dataset(name, dataset_): def register_image_dataset(name, dataset):
"""Registers a new image dataset. """Registers a new image dataset.
Args: Args:
name (str): key corresponding to the new dataset. name (str): key corresponding to the new dataset.
dataset_ (Dataset): the new dataset class. dataset (Dataset): the new dataset class.
Examples:: Examples::
@ -67,18 +69,20 @@ def register_image_dataset(name, dataset_):
sources=['new_dataset', 'dukemtmcreid'] sources=['new_dataset', 'dukemtmcreid']
) )
""" """
global __image_datasets
curr_datasets = list(__image_datasets.keys()) curr_datasets = list(__image_datasets.keys())
if name in curr_datasets: if name in curr_datasets:
raise ValueError('The given name already exists, please choose another name excluding {}'.format(curr_datasets)) raise ValueError('The given name already exists, please choose '
__image_datasets[name] = dataset_ 'another name excluding {}'.format(curr_datasets))
__image_datasets[name] = dataset
def register_video_dataset(name, dataset_): def register_video_dataset(name, dataset):
"""Registers a new video dataset. """Registers a new video dataset.
Args: Args:
name (str): key corresponding to the new dataset. name (str): key corresponding to the new dataset.
dataset_ (Dataset): the new dataset class. dataset (Dataset): the new dataset class.
Examples:: Examples::
@ -96,7 +100,9 @@ def register_video_dataset(name, dataset_):
sources=['new_dataset', 'ilidsvid'] sources=['new_dataset', 'ilidsvid']
) )
""" """
global __video_datasets
curr_datasets = list(__video_datasets.keys()) curr_datasets = list(__video_datasets.keys())
if name in curr_datasets: if name in curr_datasets:
raise ValueError('The given name already exists, please choose another name excluding {}'.format(curr_datasets)) raise ValueError('The given name already exists, please choose '
__video_datasets[name] = dataset_ 'another name excluding {}'.format(curr_datasets))
__video_datasets[name] = dataset