yolo_v3_param - 3.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
model_type : YOLOv3
  yolo_v3_param {
    num_classes: 20
    anchorCnt: 3
    layer_name: "59"
    layer_name: "67"
    layer_name: "75"
    conf_threshold: 0.3
    nms_threshold: 0.45
    biases: 10
    biases: 13
    biases: 16
    biases: 30
    biases: 33
    biases: 23
    biases: 30
    biases: 61
    biases: 62
    biases: 45
    biases: 59
    biases: 119
    biases: 116
    biases: 90
    biases: 156
    biases: 198
    biases: 373
    biases: 326
    test_mAP: false
  }

下表中列出了 YOLOv3 模型的参数。您可按自己的要求来修改这些参数。

表 1. YOLOv3 模型参数
参数类型 描述
num_classes 表示该模型的检测类别的数量。
anchorCnt 表示该模型的锚点的数量。
layer_name 表示内核的输出层的名称。如果模型有多个输出,请使用该参数来确保所需的顺序正确无误。请确保名称与内核中的名称相同。(如果您输入的名称无效,那么模型创建器会使用内核默认顺序。)
conf_threshold 框置信度的阈值,可修改以适应您的实际应用。
nms_threshold NMS 的阈值。
biases 这些参数与模型参数相同。每个偏差都写入单独一行。(Biases amount) = anchorCnt * (output-node amount) * 2,即,(偏差量) = 锚点数 * (输出节点量) * 2。在 prototxt 中设置正确的行数。
test_mAP 如果您的模型是以 letterbox 训练的,并且您想要测试其 mAP,请将此项设为 TRUE。默认情况下,它设为 FALSE,以便加速执行。