Inheritance diagram for fastreid.data.datasets.lup.LUP:
Public Member Functions | |
| __init__ (self, root='datasets', sub_name='lmdb_300_30', **kwargs) | |
| get_num_pids (self, data) | |
| get_num_cams (self, data) | |
| parse_data (self, data) | |
| show_test (self) | |
 Public Member Functions inherited from fastreid.data.datasets.bases.ImageDataset | |
| show_train (self) | |
| Public Member Functions inherited from fastreid.data.datasets.bases.Dataset | |
| __getitem__ (self, index) | |
| __len__ (self) | |
| __radd__ (self, other) | |
| show_summary (self) | |
| combine_all (self) | |
| check_before_run (self, required_files) | |
Public Attributes | |
| base_root | |
| sub_name | |
| data_root | |
| lmdb_path | |
| key_path | |
| train | |
| pids | |
| Public Attributes inherited from fastreid.data.datasets.bases.ImageDataset | |
| query | |
| gallery | |
| Public Attributes inherited from fastreid.data.datasets.bases.Dataset | |
| train | |
| query | |
| gallery | |
| transform | |
| mode | |
| combineall | |
| verbose | |
| num_train_pids | |
| num_train_cams | |
| data | |
Static Public Attributes | |
| str | dir_name = 'lup' |
Additional Inherited Members | |
| Static Protected Attributes inherited from fastreid.data.datasets.bases.Dataset | |
| list | _junk_pids = [] |
Detailed Description
Constructor & Destructor Documentation
◆ __init__()
| fastreid.data.datasets.lup.LUP.__init__ | ( | self, | |
| root = 'datasets', | |||
| sub_name = 'lmdb_300_30', | |||
| ** | kwargs ) |
Reimplemented from fastreid.data.datasets.bases.ImageDataset.
Definition at line 19 of file lup.py.
19 def __init__(self, root='datasets', sub_name='lmdb_300_30', **kwargs):
20 self.base_root = root
21 self.sub_name = sub_name
22 self.data_root = os.path.join(self.base_root, self.dir_name, self.sub_name)
23 self.lmdb_path = os.path.join(self.data_root, 'lmdb')
24 self.key_path = os.path.join(self.data_root, 'keys.pkl')
25
26 required_files = [self.data_root, self.lmdb_path, self.key_path]
27 self.check_before_run(required_files)
28
29 with open(self.key_path, 'rb') as f:
30 data = pickle.load(f)
31
32 self.train = []
33 self.pids = set()
34 for key, pid in zip(data['keys'], data['pids']):
35 self.train.append([key, pid, 0])
36 self.pids.add(pid)
37 self.pids = sorted(list(self.pids))
38
39 super(LUP, self).__init__(self.train, [], [], **kwargs)
40
41
Member Function Documentation
◆ get_num_cams()
| fastreid.data.datasets.lup.LUP.get_num_cams | ( | self, | |
| data ) |
Returns the number of training cameras.
Reimplemented from fastreid.data.datasets.bases.Dataset.
Definition at line 45 of file lup.py.
45 def get_num_cams(self, data):
46 return 1
47
◆ get_num_pids()
| fastreid.data.datasets.lup.LUP.get_num_pids | ( | self, | |
| data ) |
Returns the number of training person identities.
Reimplemented from fastreid.data.datasets.bases.Dataset.
Definition at line 42 of file lup.py.
42 def get_num_pids(self, data):
43 return len(self.pids)
44
◆ parse_data()
| fastreid.data.datasets.lup.LUP.parse_data | ( | self, | |
| data ) |
Parses data list and returns the number of person IDs
and the number of camera views.
Args:
data (list): contains tuples of (img_path(s), pid, camid)
Reimplemented from fastreid.data.datasets.bases.Dataset.
Definition at line 48 of file lup.py.
48 def parse_data(self, data):
49 return len(self.pids), 1
50
◆ show_test()
| fastreid.data.datasets.lup.LUP.show_test | ( | self | ) |
Reimplemented from fastreid.data.datasets.bases.ImageDataset.
Definition at line 51 of file lup.py.
51 def show_test(self):
52 raise Exception('LUPerson dataset has no test split currently.')
Member Data Documentation
◆ base_root
◆ data_root
◆ dir_name
◆ key_path
◆ lmdb_path
◆ pids
◆ sub_name
◆ train
The documentation for this class was generated from the following file:
- smreid/fastreid/data/datasets/lup.py
