gpt4 book ai didi

machine-learning - 更快的 R-CNN : How to avoid multiple detections in same area?

转载 作者:行者123 更新时间:2023-11-30 08:46:17 27 4
gpt4 key购买 nike

我使用 Tensorflow 对象检测 API 从头开始​​训练 Pascal VOC 数据集。我刚刚查看了 200k 训练步骤后的第一个结果,结果还不错,尽管我经常在重叠区域中检测到许多同一类。例如,考虑以下检测(忽略第一张图像中错误的人物检测):

Multiple detection of the same motorcycle Multiple detections of the same aeroplane

是否有通用方法可以避免对同一对象进行多次检测?我猜这是由重叠的区域提案引起的,检测网络预测的对象符合高于 0.7 IoU 阈值的真实数据,所以也许将这个阈值设置得更高一点会有所帮助?

最佳答案

您的意思是从头开始,还是使用了fine_tune_checkpoint?

正如其他人所建议的,您可以增加 NMS 的 IOU 阈值(将其减少到 0.01 之类的值将有效防止某个区域与另一个区域重叠时被检测到)。

关于machine-learning - 更快的 R-CNN : How to avoid multiple detections in same area?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45691417/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com