如何在Faster-RCNN中修改匹配器的协议

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。