器件型号: TDA4AEN-Q1
我在 DeepLab ONNX 模型的编译过程中遇到了量化失败的问题。 当过程达到量化阶段时、日志指示编译已成功完成、但我无法在相应的文件夹中找到输出文件。
我采用 OSRT 进行编译、相应的配置参数如下所示。
"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,
}
),

