search

msracver / Relation-Networks-for-Object-Detection

Relation Networks for Object Detection
MIT License
1.09k stars 190 forks source link

Usage of NMS #19

Closed gudovskiy closed 5 years ago

gudovskiy commented 5 years ago

Hi, thanks for your work! I have a question about greedy-NMS usage: do you completely replace NMS with your duplicate removal network or add it complementary to greedy-NMS? From Table 1 it seems that greedy-NMS is still a part of pipeline...

chengdazhi commented 5 years ago

No, we don't use greedy-NMS when we use learn nms module. Please refer to the code for details, specifically relation_rcnn/core/tester.py, that's where greedy nms is executed when we don't use learn nms.

gudovskiy commented 5 years ago

@chengdazhi thank you!