quantize-inc.Dataset Class Reference

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

Public Attributes
	root
	size
	dim
	mean
	std
	swapRB
	toFP32
	image_list

Static Public Attributes
	models

Detailed Description

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

Constructor & Destructor Documentation

__init__()

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__()

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, transform.Normalize.mean, PyPreBlob.PyPreBlob.mean, PyClassificationDNN.PyClassificationDNN.mean, PyDetectionDNN.PyDetectionDNN.mean, PyEmotion.PyEmotion.mean, jevois::BoundedBuffer< cv::Mat, jevois::BlockingBehavior::Block, jevois::BlockingBehavior::Block >.size(), jevois::BoundedBuffer< jevois::RawImage, BlockingBehavior::Block, BlockingBehavior::Block >.size(), jevois::BoundedBuffer< typename T, BlockingBehavior WhenFull, BlockingBehavior WhenEmpty >.size(), jevois::VideoBuffers.size(), quantize-inc.Dataset.size, transform.Resize.size, transform.CenterCrop.size, mobilenet_v1.MobileNetV1.std, mobilenet_v2.MobileNetV2.std, quantize-inc.Dataset.std, transform.Normalize.std, quantize-inc.Dataset.swapRB, and quantize-inc.Dataset.toFP32.

__len__()

quantize-inc.Dataset.__len__

(

self

)

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

References quantize-inc.Dataset.image_list.

load_image_list()

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

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

Member Data Documentation

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__().

mean

quantize-inc.Dataset.mean

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

Referenced by transform.Normalize.__call__(), quantize-inc.Dataset.__getitem__(), mobilenet_v1.MobileNetV1._preprocess(), mobilenet_v2.MobileNetV2._preprocess(), PyPreBlob.PyPreBlob.process(), PyClassificationDNN.PyClassificationDNN.process(), and PyDetectionDNN.PyDetectionDNN.process().

models

quantize-inc.Dataset.models

static

Initial value:

= dict(
    mobilenetv1=Quantize(model_path='../../models/image_classification_mobilenet/image_classification_mobilenetv1_2022apr.onnx',
                         config_path='./inc_configs/mobilenet.yaml'),
    mobilenetv2=Quantize(model_path='../../models/image_classification_mobilenet/image_classification_mobilenetv2_2022apr.onnx',
                         config_path='./inc_configs/mobilenet.yaml'),
    mp_palmdet=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)),
    lpd_yunet=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 73 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__(), quantize-inc.Dataset.__getitem__(), mobilenet_v1.MobileNetV1._preprocess(), and mobilenet_v2.MobileNetV2._preprocess().

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