JeVoisBase  1.18
JeVois Smart Embedded Machine Vision Toolkit Base Modules
Share this page:
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__()

◆ __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

◆ 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

◆ 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

◆ std

quantize-inc.Dataset.std

◆ 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: