| { |
| "input": "/data/huichiwen/dev/cp/onnx_model/pointpillars.onnx", |
| "output_dir": "./output", |
| "output_name": "centerpoint.axmodel", |
| "model_type": "ONNX", |
| "target_hardware": "AX650", |
| "npu_mode": "NPU3", |
|
|
| "input_shapes": "input.1:1x10x30000x20;indices_input:1x30000x2", |
|
|
| "quant": { |
| "input_configs": [ |
| { |
| "tensor_name": "input.1", |
| "calibration_dataset": "/data/huichiwen/dev/cp/quant_data/features.tar.gz", |
| "calibration_format": "Numpy", |
| "calibration_size": 32 |
| }, |
| { |
| "tensor_name": "indices_input", |
| "calibration_dataset": "/data/huichiwen/dev/cp/quant_data/indices.tar.gz", |
| "calibration_format": "Numpy", |
| "calibration_size": 32 |
| } |
| ], |
| "calibration_method": "MinMax", |
| "enable_smooth_quant": false, |
| "smooth_quant_strength": 0.5, |
| "layer_configs": [ |
| { |
| "layer_names": [ |
| "pfe_/pfn_layers.0/linear/MatMul", |
| "pfe_/pfn_layers.0/norm/BatchNormalization", |
| "pfe_/pfn_layers.0/Relu", |
| "pfe_/pfn_layers.0/ReduceMax", |
| "pfe_/pfn_layers.1/linear/MatMul", |
| "pfe_/pfn_layers.1/norm/BatchNormalization", |
| "pfe_/pfn_layers.1/Relu", |
| "pfe_/pfn_layers.1/ReduceMax", |
| "ScatterND_1" |
|
|
| ], |
| "data_type": "FP32", |
| "output_data_type": "FP32" |
| }, |
| { |
| "layer_names": [ |
| "/neck/blocks.0/blocks.0.1/Conv", |
| "/neck/blocks.0/blocks.0.3/Relu", |
| "/neck/blocks.0/blocks.0.4/Conv", |
| "/neck/blocks.0/blocks.0.6/Relu", |
| "/neck/blocks.0/blocks.0.7/Conv", |
| "/neck/blocks.0/blocks.0.9/Relu", |
| "/neck/blocks.0/blocks.0.10/Conv", |
| "/neck/blocks.0/blocks.0.12/Relu", |
| "/neck/deblocks.0/deblocks.0.0/Conv", |
| "/neck/blocks.1/blocks.1.1/Conv", |
| "/neck/deblocks.0/deblocks.0.2/Relu", |
| "/neck/blocks.1/blocks.1.3/Relu", |
| "/neck/blocks.1/blocks.1.4/Conv", |
| "/neck/blocks.1/blocks.1.6/Relu", |
| "/neck/blocks.1/blocks.1.7/Conv", |
| "/neck/blocks.1/blocks.1.9/Relu", |
| "/neck/blocks.1/blocks.1.10/Conv", |
| "/neck/blocks.1/blocks.1.12/Relu", |
| "/neck/blocks.1/blocks.1.13/Conv", |
| "/neck/blocks.1/blocks.1.15/Relu", |
| "/neck/blocks.1/blocks.1.16/Conv", |
| "/neck/blocks.1/blocks.1.18/Relu", |
| "/neck/deblocks.1/deblocks.1.0/Conv", |
| "/neck/deblocks.1/deblocks.1.2/Relu", |
| "/neck/blocks.2/blocks.2.1/Conv", |
| "/neck/blocks.2/blocks.2.3/Relu", |
| "/neck/blocks.2/blocks.2.4/Conv", |
| "/neck/blocks.2/blocks.2.6/Relu", |
| "/neck/blocks.2/blocks.2.7/Conv", |
| "/neck/blocks.2/blocks.2.9/Relu", |
| "/neck/blocks.2/blocks.2.10/Conv", |
| "/neck/blocks.2/blocks.2.12/Relu", |
| "/neck/blocks.2/blocks.2.13/Conv", |
| "/neck/blocks.2/blocks.2.15/Relu", |
| "/neck/blocks.2/blocks.2.16/Conv", |
| "/neck/blocks.2/blocks.2.18/Relu" |
| ], |
| "data_type": "FP32", |
| "output_data_type": "FP32" |
| }, |
| { |
| "layer_names": [ |
| "/Exp", |
| "/Sigmoid", |
| "/Exp_1", |
| "/Sigmoid_1", |
| "/Exp_2", |
| "/Sigmoid_2", |
| "/Exp_3", |
| "/Sigmoid_3", |
| "/Exp_4", |
| "/Sigmoid_4", |
| "/Exp_5", |
| "/Sigmoid_5", |
| "/ReduceMax", |
| "/ReduceMax_1", |
| "/ReduceMax_2", |
| "/ReduceMax_3", |
| "/ReduceMax_4", |
| "/ReduceMax_5" |
| |
| ], |
| "data_type": "FP32", |
| "output_data_type": "FP32" |
| } |
| ] |
| }, |
| "input_processors": [ |
| { |
| "tensor_name": "input.1", |
| "src_dtype": "FP32" |
| }, |
| { |
| "tensor_name": "indices_input", |
| "src_dtype": "S32" |
| } |
| ], |
|
|
| "compiler": { |
| "check": 0 |
| } |
| } |
|
|
