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
0ee5654f
Commit
0ee5654f
authored
Oct 11, 2018
by
Kai Chen
Browse files
minor fix for fast rcnn
parent
8e5bfd8b
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
7 deletions
+6
-7
configs/fast_mask_rcnn_r50_fpn_1x.py
configs/fast_mask_rcnn_r50_fpn_1x.py
+3
-3
configs/fast_rcnn_r50_fpn_1x.py
configs/fast_rcnn_r50_fpn_1x.py
+3
-3
mmdet/models/detectors/two_stage.py
mmdet/models/detectors/two_stage.py
+0
-1
No files found.
configs/fast_mask_rcnn_r50_fpn_1x.py
View file @
0ee5654f
...
...
@@ -74,7 +74,7 @@ data = dict(
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
size_divisor
=
32
,
proposal_file
=
data_root
+
'proposals/
train2017_r50_fpn_rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn_r50_fpn_1x_train2017
.pkl'
,
flip_ratio
=
0.5
,
with_mask
=
True
,
with_crowd
=
True
,
...
...
@@ -85,7 +85,7 @@ data = dict(
img_prefix
=
data_root
+
'val2017/'
,
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
proposal_file
=
data_root
+
'proposals/
val2017
_r50_fpn_
rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn
_r50_fpn_
1x_val2017
.pkl'
,
size_divisor
=
32
,
flip_ratio
=
0
,
with_mask
=
True
,
...
...
@@ -97,7 +97,7 @@ data = dict(
img_prefix
=
data_root
+
'val2017/'
,
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
proposal_file
=
data_root
+
'proposals/
val2017
_r50_fpn_
rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn
_r50_fpn_
1x_val2017
.pkl'
,
size_divisor
=
32
,
flip_ratio
=
0
,
with_mask
=
False
,
...
...
configs/fast_rcnn_r50_fpn_1x.py
View file @
0ee5654f
...
...
@@ -60,7 +60,7 @@ data = dict(
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
size_divisor
=
32
,
proposal_file
=
data_root
+
'proposals/
train2017_r50_fpn_rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn_r50_fpn_1x_train2017
.pkl'
,
flip_ratio
=
0.5
,
with_mask
=
False
,
with_crowd
=
True
,
...
...
@@ -71,7 +71,7 @@ data = dict(
img_prefix
=
data_root
+
'val2017/'
,
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
proposal_file
=
data_root
+
'proposals/
val2017
_r50_fpn_
rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn
_r50_fpn_
1x_val2017
.pkl'
,
size_divisor
=
32
,
flip_ratio
=
0
,
with_mask
=
False
,
...
...
@@ -83,7 +83,7 @@ data = dict(
img_prefix
=
data_root
+
'val2017/'
,
img_scale
=
(
1333
,
800
),
img_norm_cfg
=
img_norm_cfg
,
proposal_file
=
data_root
+
'proposals/
val2017
_r50_fpn_
rpn_1x
.pkl'
,
proposal_file
=
data_root
+
'proposals/
rpn
_r50_fpn_
1x_val2017
.pkl'
,
size_divisor
=
32
,
flip_ratio
=
0
,
with_mask
=
False
,
...
...
mmdet/models/detectors/two_stage.py
View file @
0ee5654f
...
...
@@ -140,7 +140,6 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
def
simple_test
(
self
,
img
,
img_meta
,
proposals
=
None
,
rescale
=
False
):
"""Test without augmentation."""
assert
proposals
is
None
,
"Fast RCNN hasn't been implemented."
assert
self
.
with_bbox
,
"Bbox head must be implemented."
x
=
self
.
extract_feat
(
img
)
...
...
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