SSD_param - 2.0 English

Vitis AI Library User Guide (UG1354)

Document ID
UG1354
Release Date
2022-01-20
Version
2.0 English
model_type : SSD
ssd_param :
{
    num_classes : 4
    nms_threshold : 0.4
    conf_threshold : 0.0
    conf_threshold : 0.6
    conf_threshold : 0.4
    conf_threshold : 0.3
    keep_top_k : 200
    top_k : 400
    prior_box_param {
    layer_width : 60,
    layer_height: 45,
    variances: 0.1
    variances: 0.1
    variances: 0.2
    variances: 0.2
    min_sizes: 21.0
    max_sizes: 45.0
    aspect_ratios: 2.0
    offset: 0.5
    step_width: 8.0
    step_height: 8.0
    flip: true
    clip: false
    }
}

The SSD parameters are listed in the following table. The parameters of the SSD-model include the threshold and PriorBox requirements. Refer to the SSD deploy.prototxt file for more information.

Table 1. SSD Model Parameters
Parameter Type Description
num_classes The actual number of detection categories for this model.
anchorCnt The number of anchors for this model.
conf_threshold The threshold of the boxes’ confidence. Each category can have a different threshold, but number must be equal to num_classes.
nms_threshold The threshold of NMS.
biases These parameters are same as the model parameters. Write each bias in a separate line. (Biases amount) = anchorCnt * (output-node amount) * 2. Set correct lines in the prototxt.
test_mAP If your model was trained with letterbox and you want to test its mAP, set this as TRUE. By default, it is set to FALSE for faster execution.
keep_top_k Each category of detection objects’ top K boxes.
top_k All the detection object’s top K boxes, except the background (the first category)
prior_box_param There is more than one PriorBox corresponding to different scales. You can find them in the original model (deploy.prototxt) These PriorBoxes should oppose each other.
Table 2. PriorBox Parameters
Parameter Type Description
layer_width/layer_height The input width/height of this layer. Such numbers can be computed from the net structure.
variances These numbers are used for boxes regression. These should be filled as in the original model. There should be four variances.
min_sizes/max_size Filled as the deploy.prototxt. Write each number on a separate line.
aspect_ratios The ratio (each one should be written in a separate line). By default, the first ratio is 1.0. If you set a new number here, there will be two ratios created. One of the numbers is the value that you have set, and the other number is the reciprocal of the value that you have set. For example, this parameter has only one set element, “ratios: 2.0”. The ratio vector has three numbers: 1.0, 2.0, and 0.5.
offset Normally, the PriorBox is created by each central point of the feature map, so that the offset is 0.5.
step_width/step_height Copy from the original file. If there are no such numbers there, you can use the following formula to compute them:
step_width = img_width ÷ layer_width
step_height = img_height ÷ layer_height
offset Normally, PriorBox is created by each central point of the feature map, so that the offset is 0.5.
flip Control whether to rotate the PriorBox and change the ratio of length/width.
clip Set as FALSE. If set to TRUE, the detection box coordinates will be [0, 1].