fastreid.data.datasets.veriwild.VeRiWild Class Reference

Inheritance diagram for fastreid.data.datasets.veriwild.VeRiWild:

Public Member Functions
	__init__ (self, root='datasets', query_list='', gallery_list='', **kwargs)
	process_dir (self, img_list, is_train=True)
	process_vehicle (self, vehicle_info)
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
	image_dir
	train_list
	vehicle_info
	query_list
	gallery_list
	imgid2vid
	imgid2camid
	imgid2imgpath
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-Wild"
str	dataset_name = "veriwild"

Additional Inherited Members
Static Protected Attributes inherited from fastreid.data.datasets.bases.Dataset
list	_junk_pids = []

Detailed Description

VeRi-Wild.

Reference:
    Lou et al. A Large-Scale Dataset for Vehicle Re-Identification in the Wild. CVPR 2019.

URL: `<https://github.com/PKU-IMRE/VERI-Wild>`_

Train dataset statistics:
    - identities: 30671.
    - images: 277797.

Definition at line 14 of file veriwild.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.datasets.veriwild.VeRiWild.__init__	(		self,
			root = 'datasets',
			query_list = '',
			gallery_list = '',
		**	kwargs )

Reimplemented from fastreid.data.datasets.bases.ImageDataset.

Reimplemented in fastreid.data.datasets.veriwild.SmallVeRiWild, fastreid.data.datasets.veriwild.MediumVeRiWild, and fastreid.data.datasets.veriwild.LargeVeRiWild.

Definition at line 29 of file veriwild.py.

    def __init__(self, root='datasets', query_list='', gallery_list='', **kwargs):
        self.dataset_dir = osp.join(root, self.dataset_dir)
 
        self.image_dir = osp.join(self.dataset_dir, 'images')
        self.train_list = osp.join(self.dataset_dir, 'train_test_split/train_list.txt')
        self.vehicle_info = osp.join(self.dataset_dir, 'train_test_split/vehicle_info.txt')
        if query_list and gallery_list:
            self.query_list = query_list
            self.gallery_list = gallery_list
        else:
            self.query_list = osp.join(self.dataset_dir, 'train_test_split/test_10000_query.txt')
            self.gallery_list = osp.join(self.dataset_dir, 'train_test_split/test_10000.txt')
 
        required_files = [
            self.image_dir,
            self.train_list,
            self.query_list,
            self.gallery_list,
            self.vehicle_info,
        ]
        self.check_before_run(required_files)
 
        self.imgid2vid, self.imgid2camid, self.imgid2imgpath = self.process_vehicle(self.vehicle_info)
 
        train = self.process_dir(self.train_list)
        query = self.process_dir(self.query_list, is_train=False)
        gallery = self.process_dir(self.gallery_list, is_train=False)
 
        super(VeRiWild, self).__init__(train, query, gallery, **kwargs)
 

Member Function Documentation

process_dir()

fastreid.data.datasets.veriwild.VeRiWild.process_dir	(		self,
			img_list,
			is_train = True )

Definition at line 59 of file veriwild.py.

    def process_dir(self, img_list, is_train=True):
        img_list_lines = open(img_list, 'r').readlines()
 
        dataset = []
        for idx, line in enumerate(img_list_lines):
            line = line.strip()
            vid = int(line.split('/')[0])
            imgid = line.split('/')[1]
            if is_train:
                vid = self.dataset_name + "_" + str(vid)
            dataset.append((self.imgid2imgpath[imgid], vid, int(self.imgid2camid[imgid])))
 
        assert len(dataset) == len(img_list_lines)
        return dataset
 

process_vehicle()

fastreid.data.datasets.veriwild.VeRiWild.process_vehicle	(		self,
			vehicle_info )

Definition at line 74 of file veriwild.py.

    def process_vehicle(self, vehicle_info):
        imgid2vid = {}
        imgid2camid = {}
        imgid2imgpath = {}
        vehicle_info_lines = open(vehicle_info, 'r').readlines()
 
        for idx, line in enumerate(vehicle_info_lines[1:]):
            vid = line.strip().split('/')[0]
            imgid = line.strip().split(';')[0].split('/')[1]
            camid = line.strip().split(';')[1]
            img_path = osp.join(self.image_dir, vid, imgid + '.jpg')
            imgid2vid[imgid] = vid
            imgid2camid[imgid] = camid
            imgid2imgpath[imgid] = img_path
 
        assert len(imgid2vid) == len(vehicle_info_lines) - 1
        return imgid2vid, imgid2camid, imgid2imgpath
 
 
@DATASET_REGISTRY.register()

Member Data Documentation

dataset_dir [1/2]

str fastreid.data.datasets.veriwild.VeRiWild.dataset_dir = "VERI-Wild"

static

Definition at line 26 of file veriwild.py.

dataset_dir [2/2]

fastreid.data.datasets.veriwild.VeRiWild.dataset_dir

Definition at line 30 of file veriwild.py.

dataset_name

str fastreid.data.datasets.veriwild.VeRiWild.dataset_name = "veriwild"

static

Definition at line 27 of file veriwild.py.

gallery_list

fastreid.data.datasets.veriwild.VeRiWild.gallery_list

Definition at line 37 of file veriwild.py.

image_dir

fastreid.data.datasets.veriwild.VeRiWild.image_dir

Definition at line 32 of file veriwild.py.

imgid2camid

fastreid.data.datasets.veriwild.VeRiWild.imgid2camid

Definition at line 51 of file veriwild.py.

imgid2imgpath

fastreid.data.datasets.veriwild.VeRiWild.imgid2imgpath

Definition at line 51 of file veriwild.py.

imgid2vid

fastreid.data.datasets.veriwild.VeRiWild.imgid2vid

Definition at line 51 of file veriwild.py.

query_list

fastreid.data.datasets.veriwild.VeRiWild.query_list

Definition at line 36 of file veriwild.py.

train_list

fastreid.data.datasets.veriwild.VeRiWild.train_list

Definition at line 33 of file veriwild.py.

vehicle_info

fastreid.data.datasets.veriwild.VeRiWild.vehicle_info

Definition at line 34 of file veriwild.py.

---

The documentation for this class was generated from the following file:

• smreid/fastreid/data/datasets/veriwild.py