Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
SOLOv2-pytorch
Commits
73a56fde
"vscode:/vscode.git/clone" did not exist on "2b605ab1d76db912b599b0fd7af59bf399b81d5f"
Commit
73a56fde
authored
Nov 25, 2018
by
Kai Chen
Browse files
add cascade rcnn model
parent
af55b977
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
2 deletions
+10
-2
MODEL_ZOO.md
MODEL_ZOO.md
+8
-0
mmdet/models/bbox_heads/bbox_head.py
mmdet/models/bbox_heads/bbox_head.py
+2
-2
No files found.
MODEL_ZOO.md
View file @
73a56fde
...
...
@@ -79,6 +79,14 @@ We released RPN, Faster R-CNN and Mask R-CNN models in the first version. More m
| R-50-FPN | pytorch | 1x | | | | | |
| R-50-FPN | pytorch | 2x | | | | | |
### Cascade R-CNN
| Backbone | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | box AP | Download |
|:--------:|:-------:|:-------:|:--------:|:-------------------:|:--------------:|:------:|:--------:|
| R-50-FPN | caffe | 1x | 5.0 | 0.592 | 8.1 | 40.3 | - |
| R-50-FPN | pytorch | 1x | 5.5 | 0.622 | 8.0 | 40.3 |
[
model
](
https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/cascade_rcnn_r50_fpn_1x_20181123-c019abac.pth
)
|
| R-50-FPN | pytorch | 20e | 5.5 | 0.622 | 8.0 | 41.1 |
[
model
](
https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/cascade_rcnn_r50_fpn_20e_20181123-c4cad178.pth
)
|
## Comparison with Detectron
...
...
mmdet/models/bbox_heads/bbox_head.py
View file @
73a56fde
...
...
@@ -171,10 +171,10 @@ class BBoxHead(nn.Module):
rois (Tensor): shape (n, 4) or (n, 5)
label (Tensor): shape (n, )
bbox_pred (Tensor): shape (n, 4*(#class+1)) or (n, 4)
img_
shape (Tensor): shape (3, )
img_
meta (dict): Image meta info.
Returns:
Tensor: same shape as input rois
Tensor:
Regressed bboxes, the
same shape as input rois
.
"""
assert
rois
.
size
(
1
)
==
4
or
rois
.
size
(
1
)
==
5
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment