Inheritance diagram for fastreid.data.datasets.bases.ImageDataset:

Public Member Functions
	__init__ (self, train, query, gallery, **kwargs)

	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
	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

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

Detailed Description

A base class representing ImageDataset.
All other image datasets should subclass it.
``__getitem__`` returns an image given index.
It will return ``img``, ``pid``, ``camid`` and ``img_path``
where ``img`` has shape (channel, height, width). As a result,
data in each batch has shape (batch_size, channel, height, width).

Definition at line 126 of file bases.py.

Constructor & Destructor Documentation

◆ init()

fastreid.data.datasets.bases.ImageDataset.__init__	(		self,
			train,
			query,
			gallery,
		**	kwargs )

Reimplemented from fastreid.data.datasets.bases.Dataset.

Definition at line 135 of file bases.py.

    def __init__(self, train, query, gallery, **kwargs):
        super(ImageDataset, self).__init__(train, query, gallery, **kwargs)
 

Member Function Documentation

◆ show_test()

fastreid.data.datasets.bases.ImageDataset.show_test ( self )

Reimplemented in fastreid.data.datasets.lup.LUP.

Definition at line 153 of file bases.py.

    def show_test(self):
        num_query_pids, num_query_cams = self.parse_data(self.query)
        num_gallery_pids, num_gallery_cams = self.parse_data(self.gallery)
 
        headers = ['subset', '# ids', '# images', '# cameras']
        csv_results = [
            ['query', num_query_pids, len(self.query), num_query_cams],
            ['gallery', num_gallery_pids, len(self.gallery), num_gallery_cams],
        ]
 
        # tabulate it
        table = tabulate(
            csv_results,
            tablefmt="pipe",
            headers=headers,
            numalign="left",
        )
        logger.info(f"=> Loaded {self.__class__.__name__} in csv format: \n" + colored(table, "cyan"))

◆ show_train()

fastreid.data.datasets.bases.ImageDataset.show_train ( self )

Definition at line 138 of file bases.py.

    def show_train(self):
        num_train_pids, num_train_cams = self.parse_data(self.train)
 
        headers = ['subset', '# ids', '# images', '# cameras']
        csv_results = [['train', num_train_pids, len(self.train), num_train_cams]]
 
        # tabulate it
        table = tabulate(
            csv_results,
            tablefmt="pipe",
            headers=headers,
            numalign="left",
        )
        logger.info(f"=> Loaded {self.__class__.__name__} in csv format: \n" + colored(table, "cyan"))
 

Member Data Documentation

◆ gallery

fastreid.data.datasets.bases.ImageDataset.gallery

Definition at line 160 of file bases.py.

◆ query

fastreid.data.datasets.bases.ImageDataset.query

Definition at line 159 of file bases.py.

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

smreid/fastreid/data/datasets/bases.py

Public Member Functions

Public Attributes

Additional Inherited Members