Open Code-of-Liujie opened 3 months ago
"Two_stage" is the default training setting. The problem you mentioned is due to the lack of "--two_stage" in your training script.
If you want to stick with the "non two_stage" setting, just modify output_proposals.sigmoid()
to None