使用配置文件 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

Vitis™ AI Library 提供了利用配置文件来读取模型参数的方法。它有助于简化模型参数的统一配置管理。此配置文件位于 /usr/share/vitis_ai_library/models/[model_name]/[model_name].prototxt

model
{
  name: "yolov3_voc"
  kernel {
     name: "yolov3_voc"
     mean: 0.0
     mean: 0.0
     mean: 0.0
     scale: 0.00390625
     scale: 0.00390625
     scale: 0.00390625
  }
  model_type : YOLOv3
  yolo_v3_param {
    …
  }
  is_tf: false
}
表 1. 编译模型与内核参数
模型/内核 参数类型 描述
模型 name 与 ${MODEL_NAME} 相同。
model_type 使用的模型的类型。支持以下类型。
  • CLASSIFICATION
  • DENSE_BOX
  • YOLOv3
  • SEGMENTATION
  • SSD
  • MULTI_TASK
  • TFREFINEDET
  • OPENPOSE
  • ROADLINE
  • POINTPILLARS_NUS
  • REFINEDET
  • POINTPILLARS
  • REID
  • MEDICALREFINEDET
  • FAIRMOT
  • HOURGLASS
内核 name DNNC 编译的结果。可包含额外的后缀 _0。名称包含后缀,例如,inception_v1_0。
mean 对应于模型中预定义的“BGR”平均值的三行。按“BGR”顺序来排列。
scale 对应于 RGB 归一化比例的三行。按“BGR”顺序来排列。如果模型在训练阶段没有比例值,那么该值应为 1。
is_tf 布尔值类型。如果模型是以 TensorFlow 训练的,请将该值设为 TRUE。如果采用 Caffe 或 PyTorch 模型,在 prototxt 中,该值可为空或设为 FALSE。