Tensorflow模型SSD Mobilenet检测阈值编辑

时间:2019-10-17 19:58:12

标签: tensorflow object-detection

我正在尝试在自定义数据集上训练SSD_MobileNet V1,以检测农场中的肉牛。我遵循了senddex tutorials来训练模型。现在正在工作。但是,在输出图像中,它仅在检测置信度值高于50%的对象周围绘制框。在输出图像上没有检测到的物体低于50%的检测置信度。我正在尝试将阈值设置为0.3。

为此,我从-

编辑了ssd_mobilenet_v1_pets.config文件。
argmax_matcher {
    matched_threshold: 0.5
    unmatched_threshold: 0.5
    ignore_thresholds: false
    negatives_lower_than_unmatched: true
    force_match_for_each_row: true
  }

argmax_matcher {
    matched_threshold: 0.3
    unmatched_threshold: 0.3
    ignore_thresholds: false
    negatives_lower_than_unmatched: true
    force_match_for_each_row: true
  }

然后导出推理图并使用该图。 但是,输出没有任何变化,每个检测到的对象的置信度都高于50%。有没有简单的方法可以更改阈值?

0 个答案:

没有答案