Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
130f60ac
Commit
130f60ac
authored
Nov 29, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[FasterRCNN] update resnet101 results
parent
339f2173
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
7 deletions
+10
-7
examples/FasterRCNN/README.md
examples/FasterRCNN/README.md
+4
-2
examples/FasterRCNN/config.py
examples/FasterRCNN/config.py
+6
-5
No files found.
examples/FasterRCNN/README.md
View file @
130f60ac
...
@@ -41,7 +41,7 @@ Predict on an image (and show output in a window):
...
@@ -41,7 +41,7 @@ Predict on an image (and show output in a window):
```
```
Evaluate the performance of a model and save to json.
Evaluate the performance of a model and save to json.
(
A pretrained model
can be downloaded in
[
model zoo
](
http://models.tensorpack.com/FasterRCNN
)
:
(
Pretrained models
can be downloaded in
[
model zoo
](
http://models.tensorpack.com/FasterRCNN
)
:
```
```
./train.py --evaluate output.json --load /path/to/model
./train.py --evaluate output.json --load /path/to/model
```
```
...
@@ -57,8 +57,10 @@ MaskRCNN results contain both bbox and segm mAP.
...
@@ -57,8 +57,10 @@ MaskRCNN results contain both bbox and segm mAP.
| Res50 | 256 | (600, 1024) | 34.4 | 49h on 8 TitanX |
| Res50 | 256 | (600, 1024) | 34.4 | 49h on 8 TitanX |
| Res50 | 512 | (800, 1333) | 35.6 | 55h on 8 P100|
| Res50 | 512 | (800, 1333) | 35.6 | 55h on 8 P100|
| Res50 | 512 | (800, 1333) | 36.9/32.3 | 59h on 8 P100|
| Res50 | 512 | (800, 1333) | 36.9/32.3 | 59h on 8 P100|
| Res101 | 512 | (800, 1333) | 40.1/34.4 | 70h on 8 P100|
Note that these models are trained with a longer learning schedule than the paper.
Note that these models are trained with a longer learning schedule than the paper,
and get about 1mAP better performance.
## Notes
## Notes
...
...
examples/FasterRCNN/config.py
View file @
130f60ac
...
@@ -5,22 +5,23 @@
...
@@ -5,22 +5,23 @@
import
numpy
as
np
import
numpy
as
np
# mode flags ---------------------
# mode flags ---------------------
MODE_MASK
=
Fals
e
MODE_MASK
=
Tru
e
# dataset -----------------------
# dataset -----------------------
BASEDIR
=
'/path/to/your/COCO/DIR'
BASEDIR
=
'/path/to/your/COCO/DIR'
TRAIN_DATASET
=
[
'train2014'
,
'valminusminival2014'
]
TRAIN_DATASET
=
[
'train2014'
,
'valminusminival2014'
]
VAL_DATASET
=
'minival2014'
# only support evaluation on
on
e dataset
VAL_DATASET
=
'minival2014'
# only support evaluation on
singl
e dataset
NUM_CLASS
=
81
NUM_CLASS
=
81
CLASS_NAMES
=
[]
# NUM_CLASS strings
CLASS_NAMES
=
[]
# NUM_CLASS strings
# basemodel ----------------------
# basemodel ----------------------
RESNET_NUM_BLOCK
=
[
3
,
4
,
6
,
3
]
# resnet50
RESNET_NUM_BLOCK
=
[
3
,
4
,
6
,
3
]
# resnet50
# RESNET_NUM_BLOCK = [3, 4, 23, 3] # resnet101
# preprocessing --------------------
# preprocessing --------------------
SHORT_EDGE_SIZE
=
6
00
SHORT_EDGE_SIZE
=
8
00
MAX_SIZE
=
1
024
MAX_SIZE
=
1
333
# alternative (
better) setting: 800, 1333
# alternative (
worse & faster) setting: 600, 1024
# anchors -------------------------
# anchors -------------------------
ANCHOR_STRIDE
=
16
ANCHOR_STRIDE
=
16
...
...
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