fastreid.data.common.CommDataset Class Reference

Inheritance diagram for fastreid.data.common.CommDataset:

Public Member Functions
	__init__ (self, img_items, transform=None, relabel=True)
	__len__ (self)
	__getitem__ (self, index)
	num_classes (self)
	num_cameras (self)

Public Attributes
	img_items
	transform
	relabel
	pids
	cams
	pid_dict
	cam_dict

Detailed Description

Image Person ReID Dataset

Definition at line 15 of file common.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.common.CommDataset.__init__	(		self,
			img_items,
			transform = None,
			relabel = True )

Definition at line 18 of file common.py.

    def __init__(self, img_items, transform=None, relabel=True):
        self.img_items = img_items
        self.transform = transform
        self.relabel = relabel
 
        pid_set = set()
        cam_set = set()
        for i in img_items:
            pid_set.add(i[1])
            cam_set.add(i[2])
 
        self.pids = sorted(list(pid_set))
        self.cams = sorted(list(cam_set))
        if relabel:
            self.pid_dict = dict([(p, i) for i, p in enumerate(self.pids)])
            self.cam_dict = dict([(p, i) for i, p in enumerate(self.cams)])
 

Member Function Documentation

__getitem__()

fastreid.data.common.CommDataset.__getitem__	(		self,
			index )

Definition at line 38 of file common.py.

    def __getitem__(self, index):
        img_path, pid, camid = self.img_items[index]
        img = read_image(img_path)
        if self.transform is not None: img = self.transform(img)
        if self.relabel:
            pid = self.pid_dict[pid]
            camid = self.cam_dict[camid]
        return {
            "images": img,
            "targets": pid,
            "camids": camid,
            "img_paths": img_path,
        }
 

__len__()

fastreid.data.common.CommDataset.__len__

(

self

)

Definition at line 35 of file common.py.

    def __len__(self):
        return len(self.img_items)
 

num_cameras()

fastreid.data.common.CommDataset.num_cameras

(

self

)

Definition at line 57 of file common.py.

    def num_cameras(self):
        return len(self.cams)
 

num_classes()

fastreid.data.common.CommDataset.num_classes

(

self

)

Definition at line 53 of file common.py.

    def num_classes(self):
        return len(self.pids)
 

Member Data Documentation

cam_dict

fastreid.data.common.CommDataset.cam_dict

Definition at line 33 of file common.py.

cams

fastreid.data.common.CommDataset.cams

Definition at line 30 of file common.py.

img_items

fastreid.data.common.CommDataset.img_items

Definition at line 19 of file common.py.

pid_dict

fastreid.data.common.CommDataset.pid_dict

Definition at line 32 of file common.py.

pids

fastreid.data.common.CommDataset.pids

Definition at line 29 of file common.py.

relabel

fastreid.data.common.CommDataset.relabel

Definition at line 21 of file common.py.

transform

fastreid.data.common.CommDataset.transform

Definition at line 20 of file common.py.

---

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

• smreid/fastreid/data/common.py