我正在尝试在自定义数据集上训练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%。有没有简单的方法可以更改阈值?