【问题标题】:tensorflowjs object detection coco ssd config file max_total_detections explanationtensorflowjs 对象检测 coco ssd 配置文件 max_total_detections 解释
【发布时间】:2020-06-25 11:16:16
【问题描述】:
有人可以快速解释或确认我的猜测,即:
max_detections_per_class: 100
max_total_detections: 100
在我的情况下ssdlite_mobilenet_v2_coco.config
在第 134 和 135 行。
从预测的原始输出来看,我的猜测是预测总是“尝试”检测图像中的 100 个对象,尽管图像中有实际的对象数量。假设只有一只猫,在我返回的原始预测数据中仍然会检测到 100 个对象。如果模型训练正确,当然应该只有一个高分预测。
正确吗?
谢谢!
【问题讨论】:
标签:
tensorflow
object-detection
tensorflow.js
【解决方案1】:
是的,你是对的。
它将尝试检测 100 个对象。
然后将对这 100 个检测结果进行分类,其中只有一个应该被正确识别为猫。
但这也取决于您的非最大抑制配置。如果 NMS 的得分阈值较低且 IoU 较高,则它可以显示对猫的多次检测(我的意思是重叠检测)。
你可以弄乱这些值,但从已发表的论文来看,每张图像的最大检测数应该总是比图像中的实际对象多约 3 倍。
根据我的经验,即使在我的数据中图像中的对象较少(例如,每张图像 1 个),我也会获得更好的结果。