Skip to content

Commit

Permalink
Fix paper regex (open-mmlab#354)
Browse files Browse the repository at this point in the history
* Fix paper regex

Signed-off-by: lizz <lizz@sensetime.com>

* Fix lint

Signed-off-by: lizz <lizz@sensetime.com>
  • Loading branch information
innerlee authored Dec 16, 2020
1 parent d70cbf7 commit 7bd4484
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 7 deletions.
2 changes: 1 addition & 1 deletion docs/stat.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

# count papers
papers = set(x.lower().strip()
for x in re.findall(r'\btitle={(.*)}', content))
for x in re.findall(r'\btitle\s*=\s*{(.*)}', content))
paperlist = '\n'.join(sorted(' - ' + x for x in papers))
# count configs
configs = set(x.lower().strip()
Expand Down
7 changes: 4 additions & 3 deletions mmpose/models/backbones/mspn.py
Original file line number Diff line number Diff line change
Expand Up @@ -382,9 +382,9 @@ def forward(self, img):

@BACKBONES.register_module()
class MSPN(BaseBackbone):
"""MSPN backbone.
Paper ref: Li et al. "Rethinking on Multi-Stage Networks
for Human Pose Estimation" (CVPR 2020).
"""MSPN backbone. Paper ref: Li et al. "Rethinking on Multi-Stage Networks
for Human Pose Estimation" (CVPR 2020).
Args:
unit_channels (int): Number of Channels in an upsample unit.
Default: 256
Expand All @@ -398,6 +398,7 @@ class MSPN(BaseBackbone):
Default: dict(type='BN')
res_top_channels (int): Number of channels of feature from ResNetTop.
Default: 64.
Example:
>>> from mmpose.models import MSPN
>>> import torch
Expand Down
7 changes: 4 additions & 3 deletions mmpose/models/backbones/rsn.py
Original file line number Diff line number Diff line change
Expand Up @@ -509,9 +509,10 @@ def forward(self, img):

@BACKBONES.register_module()
class RSN(BaseBackbone):
"""Residual Steps Network backbone.
Paper ref: Cai et al. "Learning Delicate Local Representations
for Multi-Person Pose Estimation" (ECCV 2020).
"""Residual Steps Network backbone. Paper ref: Cai et al. "Learning
Delicate Local Representations for Multi-Person Pose Estimation" (ECCV
2020).
Args:
unit_channels (int): Number of Channels in an upsample unit.
Default: 256
Expand Down

0 comments on commit 7bd4484

Please sign in to comment.