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 模型的参数。您可按自己的要求来修改这些参数。
参数类型 | 描述 |
---|---|
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,以便加速执行。 |