quantize-ort.Quantize Class Reference

Public Member Functions
def	__init__ (self, model_path, calibration_image_dir, transforms=Compose(), per_channel=False, act_type='int8', wt_type='int8')
def	check_opset (self, convert=True)
def	run (self)

Public Attributes
	type_dict
	calibration_image_dir
	per_channel
	act_type
	wt_type
	dr

Static Public Attributes
	yunet
	model_path
string	calibration_image_dir = '../../benchmark/data/face_detection',
	transforms = Compose([Resize(size=(160, 120))])),
	sface
	pphumenseg
	ppresnet50
	youtureid
	crnn_en
	crnn_cn

Detailed Description

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

Constructor & Destructor Documentation

__init__()

def quantize-ort.Quantize.__init__	(		self,
			model_path,
			calibration_image_dir,
			transforms = Compose(),
			per_channel = False,
			act_type = 'int8',
			wt_type = 'int8'
	)

Definition at line 43 of file quantize-ort.py.

Member Function Documentation

check_opset()

def quantize-ort.Quantize.check_opset	(		self,
			convert = True
	)

Definition at line 56 of file quantize-ort.py.

References mobilenet_v1.MobileNetV1.model_path, mobilenet_v2.MobileNetV2.model_path, mp_palmdet.MPPalmDet.model_path, lpd_yunet.LPD_YuNet.model_path, quantize-inc.Quantize.model_path, quantize-inc.Dataset.model_path, and quantize-ort.Quantize.model_path.

Referenced by quantize-ort.Quantize.run().

run()

def quantize-ort.Quantize.run

(

self

)

Definition at line 68 of file quantize-ort.py.

References quantize-ort.Quantize.act_type, quantize-ort.Quantize.check_opset(), quantize-ort.Quantize.dr, mobilenet_v2.MobileNetV2.model_path, mp_palmdet.MPPalmDet.model_path, mobilenet_v1.MobileNetV1.model_path, lpd_yunet.LPD_YuNet.model_path, quantize-inc.Quantize.model_path, quantize-inc.Dataset.model_path, quantize-ort.Quantize.model_path, quantize-ort.Quantize.per_channel, quantize-ort.Quantize.type_dict, and quantize-ort.Quantize.wt_type.

Member Data Documentation

act_type

quantize-ort.Quantize.act_type

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

Referenced by quantize-ort.Quantize.run().

calibration_image_dir [1/2]

quantize-ort.Quantize.calibration_image_dir

Definition at line 47 of file quantize-ort.py.

calibration_image_dir [2/2]

string quantize-ort.Quantize.calibration_image_dir = '../../benchmark/data/face_detection',

static

Definition at line 82 of file quantize-ort.py.

crnn_cn

quantize-ort.Quantize.crnn_cn

static

Initial value:

= Quantize(model_path='../../models/text_recognition_crnn/text_recognition_CRNN_CN_2021nov.onnx',
                     calibration_image_dir='../../benchmark/data/text',
                     transforms=Compose([Resize(size=(100, 32))]))

Definition at line 101 of file quantize-ort.py.

crnn_en

quantize-ort.Quantize.crnn_en

static

Initial value:

= Quantize(model_path='../../models/text_recognition_crnn/text_recognition_CRNN_EN_2021sep.onnx',
                     calibration_image_dir='../../benchmark/data/text',
                     transforms=Compose([Resize(size=(100, 32)), ColorConvert(ctype=cv.COLOR_BGR2GRAY)])),

Definition at line 98 of file quantize-ort.py.

dr

quantize-ort.Quantize.dr

Definition at line 54 of file quantize-ort.py.

Referenced by quantize-ort.Quantize.run().

model_path

quantize-ort.Quantize.model_path

static

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

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

per_channel

quantize-ort.Quantize.per_channel

Definition at line 49 of file quantize-ort.py.

Referenced by quantize-ort.Quantize.run().

pphumenseg

quantize-ort.Quantize.pphumenseg

static

Initial value:

= Quantize(model_path='../../models/human_segmentation_pphumanseg/human_segmentation_pphumanseg_2021oct.onnx',
                        calibration_image_dir='../../benchmark/data/human_segmentation',
                        transforms=Compose([Resize(size=(192, 192))])),

Definition at line 87 of file quantize-ort.py.

ppresnet50

quantize-ort.Quantize.ppresnet50

static

Initial value:

= Quantize(model_path='../../models/image_classification_ppresnet/image_classification_ppresnet50_2022jan.onnx',
                        calibration_image_dir='../../benchmark/data/image_classification',
                        transforms=Compose([Resize(size=(224, 224))])),

Definition at line 90 of file quantize-ort.py.

sface

quantize-ort.Quantize.sface

static

Initial value:

= Quantize(model_path='../../models/face_recognition_sface/face_recognition_sface_2021dec.onnx',
                   calibration_image_dir='../../benchmark/data/face_recognition',
                   transforms=Compose([Resize(size=(112, 112))])),

Definition at line 84 of file quantize-ort.py.

transforms

quantize-ort.Quantize.transforms = Compose([Resize(size=(160, 120))])),

static

Definition at line 83 of file quantize-ort.py.

Referenced by transform.Compose.__call__().

type_dict

quantize-ort.Quantize.type_dict

Definition at line 44 of file quantize-ort.py.

Referenced by quantize-ort.Quantize.run().

wt_type

quantize-ort.Quantize.wt_type

Definition at line 51 of file quantize-ort.py.

Referenced by quantize-ort.Quantize.run().

youtureid

quantize-ort.Quantize.youtureid

static

Initial value:

= Quantize(model_path='../../models/person_reid_youtureid/person_reid_youtu_2021nov.onnx',
                       calibration_image_dir='../../benchmark/data/person_reid',
                       transforms=Compose([Resize(size=(128, 256))])),

Definition at line 94 of file quantize-ort.py.

yunet

quantize-ort.Quantize.yunet

static

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

---

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

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