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
87059de5
Commit
87059de5
authored
May 22, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update docs
parent
a19eb489
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
7 deletions
+5
-7
examples/FasterRCNN/NOTES.md
examples/FasterRCNN/NOTES.md
+1
-3
examples/FasterRCNN/README.md
examples/FasterRCNN/README.md
+4
-4
No files found.
examples/FasterRCNN/NOTES.md
View file @
87059de5
...
@@ -47,8 +47,6 @@ Speed:
...
@@ -47,8 +47,6 @@ Speed:
a slow convolution algorithm, or you spend more time on autotune.
a slow convolution algorithm, or you spend more time on autotune.
This is a general problem of TensorFlow when running against variable-sized input.
This is a general problem of TensorFlow when running against variable-sized input.
3.
With a large roi batch size (e.g. >= 256), GPU utilitization should stay above 90%.
3.
This implementation is about 14% slower than detectron,
4.
This implementation is about 14% slower than detectron,
probably due to the lack of specialized ops (e.g. AffineChannel, ROIAlign) in TensorFlow.
probably due to the lack of specialized ops (e.g. AffineChannel, ROIAlign) in TensorFlow.
It's certainly faster than other TF implementation.
It's certainly faster than other TF implementation.
examples/FasterRCNN/README.md
View file @
87059de5
# Faster-RCNN / Mask-RCNN on COCO
# Faster-RCNN / Mask-RCNN on COCO
This example provides a minimal (only 1.6k lines) but faithful implementation
the
This example provides a minimal (only 1.6k lines) but faithful implementation
of
following papers in combination
:
the following papers
:
+
[
Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
](
https://arxiv.org/abs/1506.01497
)
+
[
Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
](
https://arxiv.org/abs/1506.01497
)
+
[
Feature Pyramid Networks for Object Detection
](
https://arxiv.org/abs/1612.03144
)
+
[
Feature Pyramid Networks for Object Detection
](
https://arxiv.org/abs/1612.03144
)
...
@@ -30,7 +30,7 @@ DIR/
...
@@ -30,7 +30,7 @@ DIR/
## Usage
## Usage
Change config in
`config.py`
:
Change config in
`config.py`
:
1.
Change
`BASEDIR`
to
`/path/to/DIR`
as described above.
1.
Change
`BASEDIR`
to
`/path/to/DIR`
as described above.
2.
Change
`MODE_MASK`
to switch Faster-RCNN or Mask-RCNN
.
2.
Change
`MODE_MASK`
/
`MODE_FPN`
, or other options you like. Recommended configurations are listed in the table below
.
Train:
Train:
```
```
...
@@ -67,7 +67,7 @@ MaskRCNN results contain both bbox and segm mAP.
...
@@ -67,7 +67,7 @@ MaskRCNN results contain both bbox and segm mAP.
|R101-C4 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100s|
|R101-C4 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100s|
|R101-C4 |512 |(800, 1333)|360k |40.8/35.1 |63h on 8 V100s|
|R101-C4 |512 |(800, 1333)|360k |40.8/35.1 |63h on 8 V100s|
The two R
-50
360k models have the same configuration __and mAP__
The two R
50-C4
360k models have the same configuration __and mAP__
as the
`R50-C4-2x`
entries in
as the
`R50-C4-2x`
entries in
[
Detectron Model Zoo
](
https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines
)
.
[
Detectron Model Zoo
](
https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines
)
.
So far this is the only TensorFlow implementation that can reproduce mAP in Detectron.
So far this is the only TensorFlow implementation that can reproduce mAP in Detectron.
...
...
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