How can softer-NMS be added to its own network structure?

[sherrygp]

PLEASE FOLLOW THESE INSTRUCTIONS BEFORE POSTING

1. Please thoroughly read README.md, INSTALL.md, GETTING_STARTED.md, and FAQ.md
2. Please search existing open and closed issues in case your issue has already been reported
3. Please try to debug the issue in case you can solve it on your own before posting

After following steps 1-3 above and agreeing to provide the detailed information requested below, you may continue with posting your issue

(Delete this line and the text above it.)

Expected results

What did you expect to see?

Actual results

What did you observe instead?

Detailed steps to reproduce

E.g.:

The command that you ran

System information

• Operating system: ?
• Compiler version: ?
• CUDA version: ?
• cuDNN version: ?
• NVIDIA driver version: ?
• GPU models (for all devices if they are not all the same): ?
• PYTHONPATH environment variable: ?
• python --version output: ?
• Anything else that seems relevant: ?

[Ethan He]

If you want to integrate softer-NMS into your own code, you can find all modifications to detectron/ at lines with flags: XYXY, PRED_STD, STD_NMS.

If you want to train your own model, Create two configs similar to these two:
First, in _init.yaml config, you need to add:

XYXY: True
PRED_STD: False
STD_NMS: False

Other staffs remain the same.

Second, in _our.yaml config, you need:

XYXY: True
PRED_STD: True
STD_NMS: True

Learning rate should be changed accordingly (see here). TRAIN.WEIGHTS should be the path to the output of _init.yaml.