quantize-inc.Dataset Class Reference

Public Member Functions
def	__init__ (self, root, size=None, dim='chw', mean=0.0, std=1.0, swapRB=False, toFP32=False)
def	load_image_list (self, path)
def	__getitem__ (self, idx)
def	__len__ (self)

Public Attributes
	root
	size
	dim
	mean
	std
	swapRB
	toFP32
	image_list

Static Public Attributes
	mobilenetv1
	model_path
	config_path
	mobilenetv2
	mp_palmdet
	lpd_yunet

Detailed Description

Definition at line 30 of file quantize-inc.py.

Constructor & Destructor Documentation

__init__()

def quantize-inc.Dataset.__init__	(		self,
			root,
			size = None,
			dim = 'chw',
			mean = 0.0,
			std = 1.0,
			swapRB = False,
			toFP32 = False
	)

Definition at line 31 of file quantize-inc.py.

Member Function Documentation

__getitem__()

def quantize-inc.Dataset.__getitem__	(		self,
			idx
	)

Definition at line 50 of file quantize-inc.py.

References quantize-inc.Dataset.dim, quantize-inc.Dataset.image_list, mobilenet_v1.MobileNetV1.mean, mobilenet_v2.MobileNetV2.mean, quantize-inc.Dataset.mean, quantize-inc.Dataset.size, mobilenet_v1.MobileNetV1.std, mobilenet_v2.MobileNetV2.std, quantize-inc.Dataset.std, quantize-inc.Dataset.swapRB, and quantize-inc.Dataset.toFP32.

__len__()

def quantize-inc.Dataset.__len__

(

self

)

Definition at line 70 of file quantize-inc.py.

References quantize-inc.Dataset.image_list.

load_image_list()

def quantize-inc.Dataset.load_image_list	(		self,
			path
	)

Definition at line 42 of file quantize-inc.py.

Member Data Documentation

config_path

quantize-inc.Dataset.config_path

static

Definition at line 75 of file quantize-inc.py.

dim

quantize-inc.Dataset.dim

Definition at line 34 of file quantize-inc.py.

Referenced by quantize-inc.Dataset.__getitem__().

image_list

quantize-inc.Dataset.image_list

Definition at line 40 of file quantize-inc.py.

Referenced by quantize-inc.Dataset.__getitem__(), and quantize-inc.Dataset.__len__().

lpd_yunet

quantize-inc.Dataset.lpd_yunet

static

Initial value:

= Quantize(model_path='../../models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may.onnx',
                       config_path='./inc_configs/lpd_yunet.yaml',
                       custom_dataset=Dataset(root='../../benchmark/data/license_plate_detection', size=(320, 240), dim='chw', toFP32=True)),

Definition at line 81 of file quantize-inc.py.

mean

quantize-inc.Dataset.mean

Definition at line 35 of file quantize-inc.py.

Referenced by transform.Normalize.__call__(), quantize-inc.Dataset.__getitem__(), PyPreBlob.PyPreBlob.process(), PyClassificationDNN.PyClassificationDNN.process(), and PyDetectionDNN.PyDetectionDNN.process().

mobilenetv1

quantize-inc.Dataset.mobilenetv1

static

Definition at line 74 of file quantize-inc.py.

mobilenetv2

quantize-inc.Dataset.mobilenetv2

static

Initial value:

= Quantize(model_path='../../models/image_classification_mobilenet/image_classification_mobilenetv2_2022apr.onnx',
                         config_path='./inc_configs/mobilenet.yaml'),

Definition at line 76 of file quantize-inc.py.

model_path

quantize-inc.Dataset.model_path

static

Definition at line 74 of file quantize-inc.py.

Referenced by quantize-ort.Quantize.check_opset(), and quantize-ort.Quantize.run().

mp_palmdet

quantize-inc.Dataset.mp_palmdet

static

Initial value:

= Quantize(model_path='../../models/palm_detection_mediapipe/palm_detection_mediapipe_2022may.onnx',
                        config_path='./inc_configs/mp_palmdet.yaml',
                        custom_dataset=Dataset(root='../../benchmark/data/palm_detection', dim='hwc', swapRB=True, mean=127.5, std=127.5, toFP32=True)),

Definition at line 78 of file quantize-inc.py.

root

quantize-inc.Dataset.root

Definition at line 32 of file quantize-inc.py.

size

quantize-inc.Dataset.size

Definition at line 33 of file quantize-inc.py.

Referenced by transform.Resize.__call__(), transform.CenterCrop.__call__(), and quantize-inc.Dataset.__getitem__().

std

quantize-inc.Dataset.std

Definition at line 36 of file quantize-inc.py.

Referenced by transform.Normalize.__call__(), and quantize-inc.Dataset.__getitem__().

swapRB

quantize-inc.Dataset.swapRB

Definition at line 37 of file quantize-inc.py.

Referenced by quantize-inc.Dataset.__getitem__().

toFP32

quantize-inc.Dataset.toFP32

Definition at line 38 of file quantize-inc.py.

Referenced by quantize-inc.Dataset.__getitem__().

---

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

• opencv_zoo/tools/quantize/quantize-inc.py