如何在faster-RCNN中修改matcher的协议

Question

我目前正在使用来自 Tensorflow 对象检测 API 的 Faster-RCNN 来训练我自己的数据。

我想改变“argmax_matcher”的正IoU值和负IoU值。我应该在 .config 文件中修改哪些字段？

非常感谢！

Answer 1

奇怪的是，argmax_matcher 不是 Faster R-CNN 的有效配置，而仅适用于 SSD。 相反，您可以在 /core/target_assigner.py 的函数 create_target_assigner() 中更改为函数调用提供的值。 您可以同时更改第一阶段：

reference == 'FasterRCNN' and stage == 'proposal'

第二个：

reference == 'FasterRCNN' and stage == 'detection'.

相关参数当然是matched_threshold和unmatched_threshold。您可能应该将force_match_for_each_row 设置为True，并注意negatives_lower_than_unmatched 默认为True。