Part Number: TDA4AEN-Q1
I encountered an issue where quantization failed during the compilation of the DeepLab ONNX model. When the process reached the quantization stage, the log indicated that the compilation was completed successfully, but I could not find the output files in the corresponding folder.
I adopt OSRT for compilation, and the corresponding configuration parameters are as follows.
"deeplabv3plus_mobilenetv2_tv_edgeailite_onnxrt_AM67A": create_model_config(
task_type="segmentation",
source=dict(),
preprocess=dict(
resize=512,
crop=512,
data_layout="NCHW",
pad_color=0,
resize_with_pad=False,
reverse_channels=False,
),
session=dict(
session_name="onnxrt",
model_path=os.path.join(
models_base_path,
"deeplab.onnx",
),
meta_arch_type=3,
input_mean=[123.675, 116.28, 103.53],
input_scale=[0.017125, 0.017507, 0.017429],
input_optimization=True,
),
postprocess=dict(with_argmax=True),
extra_info=dict(num_images=1, num_classes=2),
runtime_options={
'tensor_bits': 16,
'accuracy_level': 1,
'debug_level': 0,
'advanced_options:calibration_frames': 2,
'advanced_options:calibration_iterations': 2,
}
),
