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
6f3ed357
Commit
6f3ed357
authored
Oct 11, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update readme
parent
0ef50616
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
10 additions
and
8 deletions
+10
-8
examples/FasterRCNN/README.md
examples/FasterRCNN/README.md
+7
-7
examples/FasterRCNN/data.py
examples/FasterRCNN/data.py
+1
-1
tensorpack/tfutils/common.py
tensorpack/tfutils/common.py
+2
-0
No files found.
examples/FasterRCNN/README.md
View file @
6f3ed357
# Faster-RCNN on COCO
This example aimes to provide a minimal
Multi-GPU implementation (<1000 lines)
of ResNet50-Faster-RCNN on COCO.
This example aimes to provide a minimal
(<1000 lines) Multi-GPU implementation
of ResNet50-Faster-RCNN on COCO.
## Dependencies
+
TensorFlow nightly.
...
...
@@ -44,13 +44,13 @@ To predict on an image (and show output in a window):
The hyperparameters are not carefully tuned. You can probably get better performance by e.g. training longer.
## Files
This is a
n
minimal implementation that simply contains these files:
This is a minimal implementation that simply contains these files:
+
coco.py: load COCO data
+
data.py: prepare data for training
+
common.py:
some
common data preparation utilities
+
common.py: common data preparation utilities
+
basemodel.py: implement resnet
+
model.py: implement faster-rcnn
+
viz.py: visualization utilities
+
utils/: third-party helper functions
+
model.py: implement rpn/faster-rcnn
+
train.py: main training script
+
eval.py: utilities for evaluation
+
utils/: third-party helper functions
+
eval.py: evaluation utilities
+
viz.py: visualization utilities
examples/FasterRCNN/data.py
View file @
6f3ed357
...
...
@@ -248,7 +248,7 @@ def get_train_dataflow():
def
get_eval_dataflow
():
imgs
=
COCODetection
.
load_many
(
config
.
BASEDIR
,
config
.
VAL_DATASET
,
add_gt
=
False
)
# no filter for training
# no filter for training
ds
=
DataFromListOfDict
(
imgs
,
[
'file_name'
,
'id'
])
def
f
(
fname
):
...
...
tensorpack/tfutils/common.py
View file @
6f3ed357
...
...
@@ -46,6 +46,8 @@ def get_default_sess_config(mem_fraction=0.99):
# May hurt performance
# conf.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1
# TODO test this
# conf.graph_options.place_pruned_graph = True
return
conf
...
...
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