-
Notifications
You must be signed in to change notification settings - Fork 2.5k
proposals from RPN per image during training #676
Conversation
Thank you for your pull request and welcome to our community. We require contributors to sign our Contributor License Agreement, and we don't seem to have you on file. In order for us to review and merge your code, please sign up at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need the corporate CLA signed. If you have received this in error or have any questions, please contact us at [email protected]. Thanks! |
Thank you for signing our Contributor License Agreement. We can now accept your code for this (and any) Facebook open source project. Thanks! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @chenjoya
This is a breaking change, and I'd rather not break other people's checkpoint as is.
Could you instead add a note to the documentation, mentioning this buggy behavior?
Thanks!
Hi @fmassa , you are right, and the change of code needs to be check more. |
@fmassa Would it be okay if we add a flag (e.g., What do you think, @chenjoya ? |
@@ -158,7 +158,7 @@ def select_over_all_levels(self, boxlists): | |||
# and not per batch | |||
if self.training: | |||
objectness = torch.cat( | |||
[boxlist.get_field("objectness") for boxlist in boxlists], dim=0 | |||
[boxlist.get_field("objectness") for boxlist in boxlists], dim=0 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
just removing this extra space change and this should be good to merge!
@rodrigoberriel the documentation, after @chenjoya improvements, already makes it fairly clear about this behavior. Adding one more config argument would be an option to accommodate both behaviors, but it will just make the code more complicated. |
@fmassa fair enough about the documentation Would you be willing to review a PR trying to accommodate both behaviors? I think it would be a 5~10 lines change and we could convert this TODO into a NOTE :) maskrcnn-benchmark/maskrcnn_benchmark/modeling/rpn/inference.py Lines 154 to 159 in 1714b7c
|
@fmassa Thanks for your review. The extra space change has been removed :) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks!
@rodrigoberriel sure, why not adding an option (defaulting to the current behavior) that switches between behaviors. PRs welcome! :-) |
* proposals from RPN per image during training * README * Update README for setting FPN_POST_NMS_TOP_N_TRAIN * Update README.md * removing extra space change
Hi @fmassa , the function
select_over_all_levels
is modified for single image during training.The original issue: #672
Thank you ^ ^