axera-quan/config/config_yolov8spose_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/yolov8s-pose.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: "images"
        }
        attributes {
          input_modifications {
            dimension_permutation_preprocess {
              dimensions: 0
              dimensions: 3
              dimensions: 1
              dimensions: 2
            }
          }
        }
      }
      input_conf_items {
        selector {
        }
        attributes {
          input_modifications {
            csc {
              src_cs {
                type: CST_NV12
              }
              dst_cs {
                type: CST_RGB
              }
            }
          }
          color_space {
            type: CST_RGB
          }
        }
      }
      input_conf_items {
        attributes {
          input_modifications {
            enforce_integers {
            }
          }
          input_modifications {
            affine_preprocess {
              slope: 1.0
              bias: 0.0
              slope_divisor: 255.0
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output3"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output1"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output4"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output2"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output5"
        }
        attributes {
          output_modifications {
            dimension_permutation_postprocess {
              dimensions: 0
              dimensions: 2
              dimensions: 3
              dimensions: 1
            }
          }
        }
      }
      output_conf_items {
        selector {
          op_name: "output6"
        }
        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: 50
      batch_size: 1
      dataset_conf_items {
        path: "/root/axera/axera-quan-hjj/dataset/cocopose.tar"
        resize_conf {
        }
      }
    }
    dataset_conf_error_measurement {
      type: DATASET_TYPE_TAR
      size: 50
      batch_size: 16
      dataset_conf_items {
        path: "/root/axera/axera-quan-hjj/dataset/cocopose.tar"
        resize_conf {
        }
      }
    }
    evaluation_conf {
      path: "neuwizard.evaluator.error_measure_evaluator"
      type: EVALUATION_TYPE_ERROR_MEASURE
      source_ir_types: IR_TYPE_ONNX
      ir_types: IR_TYPE_LAVA
      score_compare_per_layer: true
    }
  }
  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/yolov8s-pose.onnx"
      type: MODEL_TYPE_ONNX
      input_tensors {
        color_space: TENSOR_COLOR_SPACE_RGB
      }
    }
  }
  output_model_items {
    model {
      path: "/root/axera/axera-quan-hjj/joint/yolov8s-pose.joint"
      type: MODEL_TYPE_JOINT
      input_tensors {
        color_space: TENSOR_COLOR_SPACE_NV12
      }
      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"