fastreid.data.datasets.shinpuhkan.Shinpuhkan Class Reference

Inheritance diagram for fastreid.data.datasets.shinpuhkan.Shinpuhkan:

Public Member Functions
	__init__ (self, root='datasets', **kwargs)
	process_train (self, train_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
	train_path
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 = "shinpuhkan"
str	dataset_name = 'shinpuhkan'

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

Detailed Description

Definition at line 16 of file shinpuhkan.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.datasets.shinpuhkan.Shinpuhkan.__init__	(		self,
			root = 'datasets',
		**	kwargs )

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

Definition at line 20 of file shinpuhkan.py.

    def __init__(self, root='datasets', **kwargs):
        self.root = root
        self.train_path = os.path.join(self.root, self.dataset_dir)
 
        required_files = [self.train_path]
        self.check_before_run(required_files)
 
        train = self.process_train(self.train_path)
 
        super().__init__(train, [], [], **kwargs)
 

Member Function Documentation

process_train()

fastreid.data.datasets.shinpuhkan.Shinpuhkan.process_train	(		self,
			train_path )

Definition at line 31 of file shinpuhkan.py.

    def process_train(self, train_path):
        data = []
 
        for root, dirs, files in os.walk(train_path):
            img_names = list(filter(lambda x: x.endswith(".jpg"), files))
            # fmt: off
            if len(img_names) == 0: continue
            # fmt: on
            for img_name in img_names:
                img_path = os.path.join(root, img_name)
                split_path = img_name.split('_')
                pid = self.dataset_name + "_" + split_path[0]
                camid = self.dataset_name + "_" + split_path[2]
                data.append((img_path, pid, camid))
 
        return data

Member Data Documentation

dataset_dir

str fastreid.data.datasets.shinpuhkan.Shinpuhkan.dataset_dir = "shinpuhkan"

static

Definition at line 17 of file shinpuhkan.py.

dataset_name

str fastreid.data.datasets.shinpuhkan.Shinpuhkan.dataset_name = 'shinpuhkan'

static

Definition at line 18 of file shinpuhkan.py.

root

fastreid.data.datasets.shinpuhkan.Shinpuhkan.root

Definition at line 21 of file shinpuhkan.py.

train_path

fastreid.data.datasets.shinpuhkan.Shinpuhkan.train_path

Definition at line 22 of file shinpuhkan.py.

---

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

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