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
}
模型/内核 | 参数类型 | 描述 |
---|---|---|
模型 | name | 与 ${MODEL_NAME} 相同。 |
model_type | 使用的模型的类型。支持以下类型。
|
|
内核 | name | DNNC 编译的结果。可包含额外的后缀 _0 。名称包含后缀,例如,inception_v1_0。 |
mean | 对应于模型中预定义的“BGR”平均值的三行。按“BGR”顺序来排列。 | |
scale | 对应于 RGB 归一化比例的三行。按“BGR”顺序来排列。如果模型在训练阶段没有比例值,那么该值应为 1。 | |
is_tf | 布尔值类型。如果模型是以 TensorFlow 训练的,请将该值设为 TRUE。如果采用 Caffe 或 PyTorch 模型,在 prototxt 中,该值可为空或设为 FALSE。 |