Commit 6f3ed357 authored by Yuxin Wu's avatar Yuxin Wu

update readme

parent 0ef50616
# Faster-RCNN on COCO # 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 ## Dependencies
+ TensorFlow nightly. + TensorFlow nightly.
...@@ -44,13 +44,13 @@ To predict on an image (and show output in a window): ...@@ -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. The hyperparameters are not carefully tuned. You can probably get better performance by e.g. training longer.
## Files ## Files
This is an minimal implementation that simply contains these files: This is a minimal implementation that simply contains these files:
+ coco.py: load COCO data + coco.py: load COCO data
+ data.py: prepare data for training + data.py: prepare data for training
+ common.py: some common data preparation utilities + common.py: common data preparation utilities
+ basemodel.py: implement resnet + basemodel.py: implement resnet
+ model.py: implement faster-rcnn + model.py: implement rpn/faster-rcnn
+ viz.py: visualization utilities
+ utils/: third-party helper functions
+ train.py: main training script + train.py: main training script
+ eval.py: utilities for evaluation + utils/: third-party helper functions
+ eval.py: evaluation utilities
+ viz.py: visualization utilities
...@@ -248,7 +248,7 @@ def get_train_dataflow(): ...@@ -248,7 +248,7 @@ def get_train_dataflow():
def get_eval_dataflow(): def get_eval_dataflow():
imgs = COCODetection.load_many(config.BASEDIR, config.VAL_DATASET, add_gt=False) 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']) ds = DataFromListOfDict(imgs, ['file_name', 'id'])
def f(fname): def f(fname):
......
...@@ -46,6 +46,8 @@ def get_default_sess_config(mem_fraction=0.99): ...@@ -46,6 +46,8 @@ def get_default_sess_config(mem_fraction=0.99):
# May hurt performance # May hurt performance
# conf.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1 # conf.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1
# TODO test this
# conf.graph_options.place_pruned_graph = True
return conf return conf
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment