config.json

{
  "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
  }
}