53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from __future__ import absolute_import
|
|
|
|
from .market1501 import Market1501
|
|
from .cuhk03 import CUHK03
|
|
from .dukemtmcreid import DukeMTMCreID
|
|
from .msmt17 import MSMT17
|
|
from .viper import VIPeR
|
|
from .grid import GRID
|
|
from .cuhk01 import CUHK01
|
|
from .prid450s import PRID450S
|
|
from .ilids import iLIDS
|
|
|
|
from .mars import Mars
|
|
from .ilidsvid import iLIDSVID
|
|
from .prid2011 import PRID2011
|
|
from .dukemtmcvidreid import DukeMTMCVidReID
|
|
|
|
|
|
__imgreid_factory = {
|
|
'market1501': Market1501,
|
|
'cuhk03': CUHK03,
|
|
'dukemtmcreid': DukeMTMCreID,
|
|
'msmt17': MSMT17,
|
|
'viper': VIPeR,
|
|
'grid': GRID,
|
|
'cuhk01': CUHK01,
|
|
'prid450s': PRID450S,
|
|
'ilids': iLIDS,
|
|
}
|
|
|
|
|
|
__vidreid_factory = {
|
|
'mars': Mars,
|
|
'ilidsvid': iLIDSVID,
|
|
'prid2011': PRID2011,
|
|
'dukemtmcvidreid': DukeMTMCVidReID,
|
|
}
|
|
|
|
|
|
def get_names():
|
|
return list(__imgreid_factory.keys()) + list(__vidreid_factory.keys())
|
|
|
|
|
|
def init_imgreid_dataset(name, **kwargs):
|
|
if name not in __imgreid_factory.keys():
|
|
raise KeyError("Invalid dataset, got '{}', but expected to be one of {}".format(name, list(__imgreid_factory.keys())))
|
|
return __imgreid_factory[name](**kwargs)
|
|
|
|
|
|
def init_vidreid_dataset(name, **kwargs):
|
|
if name not in __vidreid_factory.keys():
|
|
raise KeyError("Invalid dataset, got '{}', but expected to be one of {}".format(name, list(__vidreid_factory.keys())))
|
|
return __vidreid_factory[name](**kwargs) |