Inheritance diagram for fastreid.data.datasets.veri.VeRi:
Public Member Functions | |
| __init__ (self, root='datasets', **kwargs) | |
| process_dir (self, dir_path, is_train=True) | |
 Public Member Functions inherited from fastreid.data.datasets.bases.ImageDataset | |
| show_train (self) | |
| show_test (self) | |
| Public Member Functions inherited from fastreid.data.datasets.bases.Dataset | |
| __getitem__ (self, index) | |
| __len__ (self) | |
| __radd__ (self, other) | |
| parse_data (self, data) | |
| get_num_pids (self, data) | |
| get_num_cams (self, data) | |
| show_summary (self) | |
| combine_all (self) | |
| check_before_run (self, required_files) | |
Public Attributes | |
| dataset_dir | |
| train_dir | |
| query_dir | |
| gallery_dir | |
| 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 | dataset_dir = "veri" |
| str | dataset_name = "veri" |
Additional Inherited Members | |
| Static Protected Attributes inherited from fastreid.data.datasets.bases.Dataset | |
| list | _junk_pids = [] |
Detailed Description
VeRi.
Reference:
Liu et al. A Deep Learning based Approach for Progressive Vehicle Re-Identification. ECCV 2016.
URL: `<https://vehiclereid.github.io/VeRi/>`_
Dataset statistics:
- identities: 775.
- images: 37778 (train) + 1678 (query) + 11579 (gallery).
Constructor & Destructor Documentation
◆ __init__()
| fastreid.data.datasets.veri.VeRi.__init__ | ( | self, | |
| root = 'datasets', | |||
| ** | kwargs ) |
Reimplemented from fastreid.data.datasets.bases.ImageDataset.
Definition at line 31 of file veri.py.
31 def __init__(self, root='datasets', **kwargs):
32 self.dataset_dir = osp.join(root, self.dataset_dir)
33
34 self.train_dir = osp.join(self.dataset_dir, 'image_train')
35 self.query_dir = osp.join(self.dataset_dir, 'image_query')
36 self.gallery_dir = osp.join(self.dataset_dir, 'image_test')
37
38 required_files = [
39 self.dataset_dir,
40 self.train_dir,
41 self.query_dir,
42 self.gallery_dir,
43 ]
44 self.check_before_run(required_files)
45
46 train = self.process_dir(self.train_dir)
47 query = self.process_dir(self.query_dir, is_train=False)
48 gallery = self.process_dir(self.gallery_dir, is_train=False)
49
50 super(VeRi, self).__init__(train, query, gallery, **kwargs)
51
Member Function Documentation
◆ process_dir()
| fastreid.data.datasets.veri.VeRi.process_dir | ( | self, | |
| dir_path, | |||
| is_train = True ) |
Definition at line 52 of file veri.py.
52 def process_dir(self, dir_path, is_train=True):
53 img_paths = glob.glob(osp.join(dir_path, '*.jpg'))
54 pattern = re.compile(r'([\d]+)_c(\d\d\d)')
55
56 data = []
57 for img_path in img_paths:
58 pid, camid = map(int, pattern.search(img_path).groups())
59 if pid == -1: continue # junk images are just ignored
60 assert 1 <= pid <= 776
61 assert 1 <= camid <= 20
62 camid -= 1 # index starts from 0
63 if is_train:
64 pid = self.dataset_name + "_" + str(pid)
65 camid = self.dataset_name + "_" + str(camid)
66 data.append((img_path, pid, camid))
67
68 return data
Member Data Documentation
◆ dataset_dir [1/2]
|
static |
◆ dataset_dir [2/2]
◆ dataset_name
|
static |
◆ gallery_dir
◆ query_dir
◆ train_dir
The documentation for this class was generated from the following file:
- smreid/fastreid/data/datasets/veri.py
