fastreid.data.datasets.wildtracker.WildTrackCrop Class Reference

Inheritance diagram for fastreid.data.datasets.wildtracker.WildTrackCrop:

Public Member Functions
	__init__ (self, root='datasets', **kwargs)
	process_dir (self, dir_path)
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
	root
	dataset_dir
	train_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
	dataset_url = None
str	dataset_dir = 'Wildtrack_crop_dataset'
str	dataset_name = 'wildtrack'

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

Detailed Description

WildTrack.
Reference:
    WILDTRACK: A Multi-camera HD Dataset for Dense Unscripted Pedestrian Detection
        T. Chavdarova; P. Baqué; A. Maksai; S. Bouquet; C. Jose et al.
URL: `<https://www.epfl.ch/labs/cvlab/data/data-wildtrack/>`_
Dataset statistics:
    - identities: 313
    - images: 33979 (train only)
    - cameras: 7
Args:
    data_path(str): path to WildTrackCrop dataset
    combineall(bool): combine train and test sets as train set if True

Definition at line 15 of file wildtracker.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.datasets.wildtracker.WildTrackCrop.__init__	(		self,
			root = 'datasets',
		**	kwargs )

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

Definition at line 33 of file wildtracker.py.

    def __init__(self, root='datasets', **kwargs):
        self.root = root
        self.dataset_dir = os.path.join(self.root, self.dataset_dir)
 
        self.train_dir = os.path.join(self.dataset_dir, "crop")
 
        train = self.process_dir(self.train_dir)
        query = []
        gallery = []
 
        super(WildTrackCrop, self).__init__(train, query, gallery, **kwargs)
 

Member Function Documentation

process_dir()

fastreid.data.datasets.wildtracker.WildTrackCrop.process_dir	(		self,
			dir_path )

    :param dir_path: directory path saving images
    Returns
        data(list) = [img_path, pid, camid]

Definition at line 45 of file wildtracker.py.

    def process_dir(self, dir_path):
        r"""
        :param dir_path: directory path saving images
        Returns
            data(list) = [img_path, pid, camid]
        """
        data = []
        for dir_name in os.listdir(dir_path):
            img_lists = glob.glob(os.path.join(dir_path, dir_name, "*.png"))
            for img_path in img_lists:
                pid = self.dataset_name + "_" + dir_name
                camid = img_path.split('/')[-1].split('_')[0]
                camid = self.dataset_name + "_" + camid
                data.append([img_path, pid, camid])
        return data

Member Data Documentation

dataset_dir [1/2]

str fastreid.data.datasets.wildtracker.WildTrackCrop.dataset_dir = 'Wildtrack_crop_dataset'

static

Definition at line 30 of file wildtracker.py.

dataset_dir [2/2]

fastreid.data.datasets.wildtracker.WildTrackCrop.dataset_dir

Definition at line 35 of file wildtracker.py.

dataset_name

str fastreid.data.datasets.wildtracker.WildTrackCrop.dataset_name = 'wildtrack'

static

Definition at line 31 of file wildtracker.py.

dataset_url

fastreid.data.datasets.wildtracker.WildTrackCrop.dataset_url = None

static

Definition at line 29 of file wildtracker.py.

root

fastreid.data.datasets.wildtracker.WildTrackCrop.root

Definition at line 34 of file wildtracker.py.

train_dir

fastreid.data.datasets.wildtracker.WildTrackCrop.train_dir

Definition at line 37 of file wildtracker.py.

---

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

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