perform training on negative samples (images with no annotation)#869
perform training on negative samples (images with no annotation)#869gcheron wants to merge 1 commit intofacebookresearch:mainfrom
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 cla@fb.com. Thanks! |
|
Thank you for signing our Contributor License Agreement. We can now accept your code for this (and any) Facebook open source project. Thanks! |
|
@gcheron I have tried to train a RETINANET model with your pr_negative_samples branch. [I net_dag_utils.cc:102] Operator graph pruning prior to chain compute took: 0.000376701 secs |
|
It looks like you have done changes for RPN.py i.e. models which use RPNs will only be able to use this feature. There are on changes in "roi_data/retinanet.py" therefore this error occurs. |
@ir413 , @JiYuanFeng , this allows the use of negative samples in training when setting
TRAIN.INCLUDE_BKG_IMAGES = True(related to #697 )