fastreid.data.common.LMDBDataset Class Reference

Inheritance diagram for fastreid.data.common.LMDBDataset:

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

Public Attributes
	dataset
	transform
	relabel
	img_items
	pids
	lmdb_path
	env

Protected Member Functions
	_init_lmdb (self)

Protected Attributes
	_num_classes
	_num_cameras

Detailed Description

 Person dataset in LMDB format 

Definition at line 104 of file common.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.common.LMDBDataset.__init__	(		self,
			dataset,
			transform = None,
			relabel = False )

Definition at line 106 of file common.py.

    def __init__(self, dataset, transform=None, relabel=False):
        self.dataset   = dataset
        self.transform = transform
        self.relabel   = relabel
 
        self.img_items = dataset.train
        self.pids      = dataset.pids
 
        self._num_classes = len(self.pids)
        self._num_cameras = 1
 
        self.lmdb_path = dataset.lmdb_path
        self.env = None
 

Member Function Documentation

__getitem__()

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

Definition at line 127 of file common.py.

    def __getitem__(self, index):
        if self.env is None:
            self._init_lmdb()
        img_key, pid, camid = self.img_items[index]
        
        with self.env.begin(write=False) as txn:
            buf = txn.get(img_key.encode('ascii'))
 
        img_flat = np.frombuffer(buf, dtype=np.uint8)
        # im is in BGR order
        im = cv2.imdecode(img_flat, cv2.IMREAD_COLOR)
        im = Image.fromarray(im[:, :, ::-1])
 
        if self.transform is not None:
            im = self.transform(im)
 
        return {'images': im, 'targets': pid, 
                'camids': 0,  'img_paths': img_key}
 

__len__()

fastreid.data.common.LMDBDataset.__len__

(

self

)

Definition at line 124 of file common.py.

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

_init_lmdb()

fastreid.data.common.LMDBDataset._init_lmdb ( self )

protected

Definition at line 120 of file common.py.

    def _init_lmdb(self):
        self.env = lmdb.open(self.lmdb_path, readonly=True, lock=False, 
                        readahead=False, meminit=False)
 

num_cameras()

fastreid.data.common.LMDBDataset.num_cameras

(

self

)

Definition at line 151 of file common.py.

    def num_cameras(self):
        return self._num_cameras
    

num_classes()

fastreid.data.common.LMDBDataset.num_classes

(

self

)

Definition at line 147 of file common.py.

    def num_classes(self):
        return self._num_classes
 

Member Data Documentation

_num_cameras

fastreid.data.common.LMDBDataset._num_cameras

protected

Definition at line 115 of file common.py.

_num_classes

fastreid.data.common.LMDBDataset._num_classes

protected

Definition at line 114 of file common.py.

dataset

fastreid.data.common.LMDBDataset.dataset

Definition at line 107 of file common.py.

env

fastreid.data.common.LMDBDataset.env

Definition at line 118 of file common.py.

img_items

fastreid.data.common.LMDBDataset.img_items

Definition at line 111 of file common.py.

lmdb_path

fastreid.data.common.LMDBDataset.lmdb_path

Definition at line 117 of file common.py.

pids

fastreid.data.common.LMDBDataset.pids

Definition at line 112 of file common.py.

relabel

fastreid.data.common.LMDBDataset.relabel

Definition at line 109 of file common.py.

transform

fastreid.data.common.LMDBDataset.transform

Definition at line 108 of file common.py.

---

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

• smreid/fastreid/data/common.py