-
Notifications
You must be signed in to change notification settings - Fork 9.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Fix]: Alpha of ciou loss should not have gradient. #5835
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5835 +/- ##
==========================================
- Coverage 67.09% 67.06% -0.03%
==========================================
Files 301 301
Lines 23335 23334 -1
Branches 3841 3841
==========================================
- Hits 15657 15650 -7
Misses 6884 6884
- Partials 794 800 +6
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Suggest updating unit test |
Add a faster rcnn model with CIoULoss |
Training results of CIOU Faster R-CNN Before:
After:
Almost the same. |
CIOU with weight=12 refering from the paper
|
can be merged after update config and model link |
* [Fix]: Alpha of ciou loss does not have gradient. * add ciou faster rcnn * update loss weight
In the official implementation of CIOU loss, the
alpha
does not have gradient. And when iou<=0.5, alpha=0. Refer to:https://github.com/Zzh-tju/CIoU/blob/d8573fc9ef221e89fd0dfeef42027acd764c49ff/layers/modules/multibox_loss.py#L53