mirror of https://github.com/JDAI-CV/fast-reid.git
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
# encoding: utf-8
|
|
"""
|
|
@author: liaoxingyu
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from ...utils.registry import Registry
|
|
|
|
DATASET_REGISTRY = Registry("DATASET")
|
|
DATASET_REGISTRY.__doc__ = """
|
|
Registry for datasets
|
|
It must returns an instance of :class:`Backbone`.
|
|
"""
|
|
|
|
# Person re-id datasets
|
|
from .cuhk03 import CUHK03
|
|
from .dukemtmcreid import DukeMTMC
|
|
from .market1501 import Market1501
|
|
from .msmt17 import MSMT17
|
|
from .AirportALERT import AirportALERT
|
|
from .iLIDS import iLIDS
|
|
from .pku import PKU
|
|
from .prai import PRAI
|
|
from .sensereid import SenseReID
|
|
from .sysu_mm import SYSU_mm
|
|
from .thermalworld import Thermalworld
|
|
from .pes3d import PeS3D
|
|
from .caviara import CAVIARa
|
|
from .viper import VIPeR
|
|
from .lpw import LPW
|
|
from .shinpuhkan import Shinpuhkan
|
|
# Vehicle re-id datasets
|
|
from .veri import VeRi
|
|
from .vehicleid import VehicleID, SmallVehicleID, MediumVehicleID, LargeVehicleID
|
|
from .veriwild import VeRiWild, SmallVeRiWild, MediumVeRiWild, LargeVeRiWild
|
|
|
|
__all__ = [k for k in globals().keys() if "builtin" not in k and not k.startswith("_")]
|