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
763decc4
"tests/vscode:/vscode.git/clone" did not exist on "2bfa5a61fb21e03cb3e70b0cdace7bd8466a2817"
Commit
763decc4
authored
Oct 10, 2018
by
pangjm
Browse files
add fast rcnn api & fix minor bugs
parent
b07edd6b
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
29 additions
and
2 deletions
+29
-2
mmdet/models/detectors/__init__.py
mmdet/models/detectors/__init__.py
+2
-1
mmdet/models/detectors/fast_rcnn.py
mmdet/models/detectors/fast_rcnn.py
+25
-0
mmdet/models/detectors/two_stage.py
mmdet/models/detectors/two_stage.py
+2
-1
No files found.
mmdet/models/detectors/__init__.py
View file @
763decc4
from
.base
import
BaseDetector
from
.base
import
BaseDetector
from
.rpn
import
RPN
from
.rpn
import
RPN
from
.faster_rcnn
import
FastRCNN
from
.faster_rcnn
import
FasterRCNN
from
.faster_rcnn
import
FasterRCNN
from
.mask_rcnn
import
MaskRCNN
from
.mask_rcnn
import
MaskRCNN
__all__
=
[
'BaseDetector'
,
'RPN'
,
'FasterRCNN'
,
'MaskRCNN'
]
__all__
=
[
'BaseDetector'
,
'RPN'
,
'FastRCNN'
,
'FasterRCNN'
,
'MaskRCNN'
]
mmdet/models/detectors/fast_rcnn.py
0 → 100644
View file @
763decc4
from
.two_stage
import
TwoStageDetector
class
FastRCNN
(
TwoStageDetector
):
def
__init__
(
self
,
backbone
,
neck
,
bbox_roi_extractor
,
bbox_head
,
train_cfg
,
test_cfg
,
mask_roi_extractor
=
None
,
mask_head
=
None
,
pretrained
=
None
):
super
(
FastRCNN
,
self
).
__init__
(
backbone
=
backbone
,
neck
=
neck
,
bbox_roi_extractor
=
bbox_roi_extractor
,
bbox_head
=
bbox_head
,
train_cfg
=
train_cfg
,
test_cfg
=
test_cfg
,
mask_roi_extractor
=
mask_roi_extractor
,
mask_head
=
mask_head
,
pretrained
=
pretrained
)
mmdet/models/detectors/two_stage.py
View file @
763decc4
...
@@ -146,7 +146,8 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
...
@@ -146,7 +146,8 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
x
=
self
.
extract_feat
(
img
)
x
=
self
.
extract_feat
(
img
)
proposal_list
=
self
.
simple_test_rpn
(
proposal_list
=
self
.
simple_test_rpn
(
x
,
img_meta
,
self
.
test_cfg
.
rpn
)
if
proposals
is
None
else
proposals
x
,
img_meta
,
self
.
test_cfg
.
rpn
)
if
proposals
is
None
else
proposals
[
0
]
det_bboxes
,
det_labels
=
self
.
simple_test_bboxes
(
det_bboxes
,
det_labels
=
self
.
simple_test_bboxes
(
x
,
img_meta
,
proposal_list
,
self
.
test_cfg
.
rcnn
,
rescale
=
rescale
)
x
,
img_meta
,
proposal_list
,
self
.
test_cfg
.
rcnn
,
rescale
=
rescale
)
...
...
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