fastreid.data.datasets.vehicleid.VehicleID Class Reference

Inheritance diagram for fastreid.data.datasets.vehicleid.VehicleID:

Public Member Functions
	__init__ (self, root='datasets', test_list='', **kwargs)
	process_dir (self, list_file, 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
	image_dir
	train_list
	test_list
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 = "vehicleid"
str	dataset_name = "vehicleid"

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

Detailed Description

VehicleID.

Reference:
    Liu et al. Deep relative distance learning: Tell the difference between similar vehicles. CVPR 2016.

URL: `<https://pkuml.org/resources/pku-vehicleid.html>`_

Train dataset statistics:
    - identities: 13164.
    - images: 113346.

Definition at line 15 of file vehicleid.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.datasets.vehicleid.VehicleID.__init__	(		self,
			root = 'datasets',
			test_list = '',
		**	kwargs )

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

Reimplemented in fastreid.data.datasets.vehicleid.SmallVehicleID, fastreid.data.datasets.vehicleid.MediumVehicleID, and fastreid.data.datasets.vehicleid.LargeVehicleID.

Definition at line 30 of file vehicleid.py.

    def __init__(self, root='datasets', test_list='', **kwargs):
        self.dataset_dir = osp.join(root, self.dataset_dir)
 
        self.image_dir = osp.join(self.dataset_dir, 'image')
        self.train_list = osp.join(self.dataset_dir, 'train_test_split/train_list.txt')
        if test_list:
            self.test_list = test_list
        else:
            self.test_list = osp.join(self.dataset_dir, 'train_test_split/test_list_13164.txt')
 
        required_files = [
            self.dataset_dir,
            self.image_dir,
            self.train_list,
            self.test_list,
        ]
        self.check_before_run(required_files)
 
        train = self.process_dir(self.train_list, is_train=True)
        query, gallery = self.process_dir(self.test_list, is_train=False)
 
        super(VehicleID, self).__init__(train, query, gallery, **kwargs)
 

Member Function Documentation

process_dir()

fastreid.data.datasets.vehicleid.VehicleID.process_dir	(		self,
			list_file,
			is_train = True )

Definition at line 53 of file vehicleid.py.

    def process_dir(self, list_file, is_train=True):
        img_list_lines = open(list_file, 'r').readlines()
 
        dataset = []
        for idx, line in enumerate(img_list_lines):
            line = line.strip()
            vid = int(line.split(' ')[1])
            imgid = line.split(' ')[0]
            img_path = osp.join(self.image_dir, imgid + '.jpg')
            if is_train:
                vid = self.dataset_name + "_" + str(vid)
            dataset.append((img_path, vid, int(imgid)))
 
        if is_train: return dataset
        else:
            random.shuffle(dataset)
            vid_container = set()
            query = []
            gallery = []
            for sample in dataset:
                if sample[1] not in vid_container:
                    vid_container.add(sample[1])
                    gallery.append(sample)
                else:
                    query.append(sample)
 
            return query, gallery
 
 
@DATASET_REGISTRY.register()

Member Data Documentation

dataset_dir [1/2]

str fastreid.data.datasets.vehicleid.VehicleID.dataset_dir = "vehicleid"

static

Definition at line 27 of file vehicleid.py.

dataset_dir [2/2]

fastreid.data.datasets.vehicleid.VehicleID.dataset_dir

Definition at line 31 of file vehicleid.py.

dataset_name

str fastreid.data.datasets.vehicleid.VehicleID.dataset_name = "vehicleid"

static

Definition at line 28 of file vehicleid.py.

image_dir

fastreid.data.datasets.vehicleid.VehicleID.image_dir

Definition at line 33 of file vehicleid.py.

test_list

fastreid.data.datasets.vehicleid.VehicleID.test_list

Definition at line 36 of file vehicleid.py.

train_list

fastreid.data.datasets.vehicleid.VehicleID.train_list

Definition at line 34 of file vehicleid.py.

---

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

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