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

标签 machine-learning tensorflow object-detection

我使用 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/

相关文章:

python - mlpy - 动态时间扭曲取决于 x?

tensorflow - 如何在 Google Colab 中将 TensorFlow 2.0 设置为默认版本

amazon-web-services - TensorFlow 不使用 GPU

c++ - 在 OpenCV 中合并重叠矩形

python - 为什么PCA的数量改变了?

python - 我们应该如何对全序列进行分类?

machine-learning - 支持向量机中的 "support"是什么意思?

tensorflow - 如何保存 tf.data.Dataset 对象?

c++ - Tensorflow 对象检测 API 是否有 C++ 包装器?

tensorflow - 如何在 Tensorflow 对象检测 API 中存储最佳模型检查点,而不仅仅是最新的 5 个?