-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
[Feature] Adaptive wingloss (ICCV'19) #1072
Conversation
Codecov Report
@@ Coverage Diff @@
## dev-0.22 #1072 +/- ##
===========================================
Coverage ? 82.47%
===========================================
Files ? 203
Lines ? 16078
Branches ? 2864
===========================================
Hits ? 13260
Misses ? 2095
Partials ? 723
Flags with carried forward coverage won't be shown. Click here to find out more. Continue to review full report at Codecov.
|
@@ -159,7 +159,7 @@ def get_loss(self, output, target, target_weight): | |||
|
|||
assert not isinstance(self.loss, nn.Sequential) | |||
assert target.dim() == 4 and target_weight.dim() == 3 | |||
losses['mse_loss'] = self.loss(output, target, target_weight) | |||
losses['heatmap_loss'] = self.loss(output, target, target_weight) |
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.
Will this lead to risks of bc breaking?
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.
Maybe.
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.
OK, we can mention it in the changelog.
|
||
@LOSSES.register_module() | ||
class AdaptiveWingLoss(nn.Module): | ||
"""Adaptive wing loss. paper ref: 'Adaptive Wing Loss for Robust Face |
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.
Please follow standard reference format:
mmpose/mmpose/models/backbones/hrnet.py
Lines 218 to 219 in 228747c
`High-Resolution Representations for Labeling Pixels and Regions | |
<https://arxiv.org/abs/1904.04514>`__ |
0fd1ebc
to
25340d6
Compare
* add adaptive wingloss * move adaptive wingloss to heatmap_loss * match shape in wingloss * rename loss * mean for loss * add configs * fix unittest * rename mse_loss to heatmap_loss * add docs * fix lint Co-authored-by: ly015 <liyining0712@gmail.com>
* add adaptive wingloss * move adaptive wingloss to heatmap_loss * match shape in wingloss * rename loss * mean for loss * add configs * fix unittest * rename mse_loss to heatmap_loss * add docs * fix lint Co-authored-by: ly015 <liyining0712@gmail.com>
* add adaptive wingloss * move adaptive wingloss to heatmap_loss * match shape in wingloss * rename loss * mean for loss * add configs * fix unittest * rename mse_loss to heatmap_loss * add docs * fix lint Co-authored-by: ly015 <liyining0712@gmail.com>
* add adaptive wingloss * move adaptive wingloss to heatmap_loss * match shape in wingloss * rename loss * mean for loss * add configs * fix unittest * rename mse_loss to heatmap_loss * add docs * fix lint Co-authored-by: ly015 <liyining0712@gmail.com>
Motivation
Support Adaptive Wing Loss.
Modification
BC-breaking (Optional)
Use cases (Optional)
Checklist
Before PR:
After PR: