Tensorflow对象检测API:禁用对现成模型的非最大化抑制

Question

是否有可能和/或容易禁用Tensorflow对象检测API中提供的现成对象检测器的非极大值抑制部分？例如，我想运行提供的"SSD移动网络，它是在MSCOCO上训练的“，而不是在结束时进行非最大化操作。我怎样才能做到这一点？

Answer 1

如果您出于速度原因想要这样做，唯一的方法是编辑代码本身(请参阅arch.py#L331) --这涉及到一些问题，因为您需要用代码替换对NMS的调用，这些代码仍然会将框放入预期的输出格式。

如果您只想消除NMS的影响，您只需将配置文件的后处理部分的score_threshold和iou_threshold设置为：pets.config#L131分别为0.0和1.0，这意味着，不要过滤低得分框，而只有在它们完全重叠时才会基于iou进行剪枝框(实际上永远不会)。