axera-quan/config/config_efficientnetb0_output.prototxt

cpu_backend_settings {
  onnx_setting {
    mode: DISABLED
  }
  axe_setting {
    mode: ENABLED
    axe_param {
      optimize_slim_model: true
    }
  }
}
tasks {
  name: "task_0"
  neuwizard_conf {
    global_conf {
      target_hardware: TARGET_HARDWARE_AX620
    }
    task_conf {
      input_path: "/root/axera/axera-quan-hjj/model/efficientnet_size-160.onnx"
      output_path: "/root/axera/axera-quan-hjj/joint/model.lava_joint"
      output_type: OUTPUT_TYPE_JOINT_LAVA_AXE
    }
    operator_conf {
      input_conf_items {
        selector {
          op_name: "input"
        }
        attributes {
          input_modifications {
            dimension_permutation_preprocess {
              dimensions: 0
              dimensions: 3
              dimensions: 1
              dimensions: 2
            }
          }
        }
      }
      input_conf_items {
        selector {
        }
        attributes {
          color_space {
            type: CST_RGB
          }
        }
      }
      input_conf_items {
        attributes {
          input_modifications {
            enforce_integers {
            }
          }
          input_modifications {
            affine_preprocess {
              slope: 1.0
              bias: -0.5
              slope_divisor: 256.0
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
    }
    dataset_conf_train {
      dataset_conf_items {
        resize_conf {
        }
      }
    }
    dataset_conf_calibration {
      type: DATASET_TYPE_TAR
      size: 100
      batch_size: 16
      dataset_conf_items {
        path: "/root/axera/axera-quan-hjj/dataset/handpose.tar"
        resize_conf {
        }
      }
    }
    dataset_conf_error_measurement {
      dataset_conf_items {
        resize_conf {
        }
      }
    }
  }
  pulsar_conf {
    batch_size: 1
    ax620_virtual_npu: AX620_VIRTUAL_NPU_MODE_111
    output_dir: "/root/axera/axera-quan-hjj"
  }
  input_model_items {
    model {
      path: "/root/axera/axera-quan-hjj/model/efficientnet_size-160.onnx"
      type: MODEL_TYPE_ONNX
      input_tensors {
        color_space: TENSOR_COLOR_SPACE_RGB
      }
    }
  }
  output_model_items {
    model {
      path: "/root/axera/axera-quan-hjj/joint/efficientnetb0-handpose.joint"
      type: MODEL_TYPE_JOINT
      input_tensors {
        color_space: TENSOR_COLOR_SPACE_RGB
      }
      output_tensors {
        tensor_layout: NHWC
      }
    }
  }
  target_hardware: TARGET_HARDWARE_AX620
  cpu_backend_settings {
    onnx_setting {
      mode: DISABLED
    }
    axe_setting {
      mode: ENABLED
      axe_param {
        optimize_slim_model: true
      }
    }
  }
  cpu_affine_json {
    mode: PATH_MODE_ABS
  }
}
input_task_names: "task_0"
output_task_names: "task_0"