An abstract class representing a Dataset.
This is the base class for ``ImageDataset`` and ``VideoDataset``.
Args:
train (list): contains tuples of (img_path(s), pid, camid).
query (list): contains tuples of (img_path(s), pid, camid).
gallery (list): contains tuples of (img_path(s), pid, camid).
transform: transform function.
mode (str): 'train', 'query' or 'gallery'.
combineall (bool): combines train, query and gallery in a
dataset for training.
verbose (bool): show information.
Definition at line 16 of file bases.py.
| fastreid.data.datasets.bases.Dataset.__init__ |
( |
| self, |
|
|
| train, |
|
|
| query, |
|
|
| gallery, |
|
|
| transform = None, |
|
|
| mode = 'train', |
|
|
| combineall = False, |
|
|
| verbose = True, |
|
|
** | kwargs ) |
Reimplemented in fastreid.data.datasets.market1501.Market1501, fastreid.data.datasets.AirportALERT.AirportALERT, fastreid.data.datasets.caviara.CAVIARa, fastreid.data.datasets.cuhk_sysu.cuhkSYSU, fastreid.data.datasets.dukemtmcreid.DukeMTMC, fastreid.data.datasets.iLIDS.iLIDS, fastreid.data.datasets.lpw.LPW, fastreid.data.datasets.msmt17.MSMT17, fastreid.data.datasets.pes3d.PeS3D, fastreid.data.datasets.pku.PKU, fastreid.data.datasets.prai.PRAI, fastreid.data.datasets.sensereid.SenseReID, fastreid.data.datasets.shinpuhkan.Shinpuhkan, fastreid.data.datasets.sysu_mm.SYSU_mm, fastreid.data.datasets.thermalworld.Thermalworld, fastreid.data.datasets.vehicleid.SmallVehicleID, fastreid.data.datasets.vehicleid.MediumVehicleID, fastreid.data.datasets.vehicleid.LargeVehicleID, fastreid.data.datasets.veri.VeRi, fastreid.data.datasets.veriwild.SmallVeRiWild, fastreid.data.datasets.veriwild.MediumVeRiWild, fastreid.data.datasets.veriwild.LargeVeRiWild, fastreid.data.datasets.viper.VIPeR, fastreid.data.datasets.wildtracker.WildTrackCrop, fastreid.data.datasets.cmdm.CMDM, fastreid.data.datasets.veriwild.VeRiWild, fastreid.data.datasets.cuhk03.CUHK03, fastreid.data.datasets.lup.LUP, fastreid.data.datasets.vehicleid.VehicleID, and fastreid.data.datasets.bases.ImageDataset.
Definition at line 31 of file bases.py.
32 combineall=False, verbose=True, **kwargs):
33 self.train = train
34 self.query = query
35 self.gallery = gallery
36 self.transform = transform
37 self.mode = mode
38 self.combineall = combineall
39 self.verbose = verbose
40
41 self.num_train_pids = self.get_num_pids(self.train)
42 self.num_train_cams = self.get_num_cams(self.train)
43
44 if self.combineall:
45 self.combine_all()
46
47 if self.mode == 'train':
48 self.data = self.train
49 elif self.mode == 'query':
50 self.data = self.query
51 elif self.mode == 'gallery':
52 self.data = self.gallery
53 else:
54 raise ValueError('Invalid mode. Got {}, but expected to be '
55 'one of [train | query | gallery]'.format(self.mode))
56
