Commit bf94458d authored by Yuxin Wu's avatar Yuxin Wu

[FasterRCNN] update docs

parent dafdabf8
...@@ -21,10 +21,14 @@ This is a minimal implementation that simply contains these files: ...@@ -21,10 +21,14 @@ This is a minimal implementation that simply contains these files:
<p align="center"> <img src="https://user-images.githubusercontent.com/1381301/31527740-2f1b38ce-af84-11e7-8de1-628e90089826.png"> </p> <p align="center"> <img src="https://user-images.githubusercontent.com/1381301/31527740-2f1b38ce-af84-11e7-8de1-628e90089826.png"> </p>
3. Inference is not quite fast, because either you disable convolution autotune and end up with 3. We use ROIAlign, and because of (3), `tf.image.crop_and_resize` is NOT ROIAlign.
4. Inference is not quite fast, because either you disable convolution autotune and end up with
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.
4. In Faster-RCNN, BatchNorm statistics are not supposed to be updated during fine-tuning. 5. We only support single image per GPU for now.
6. Because of (4), BatchNorm statistics are not supposed to be updated during fine-tuning.
This specific kind of BatchNorm will need [my kernel](https://github.com/tensorflow/tensorflow/pull/12580) This specific kind of BatchNorm will need [my kernel](https://github.com/tensorflow/tensorflow/pull/12580)
which is included since TF 1.4. If using an earlier version of TF, it will be either slow or wrong. which is included since TF 1.4. If using an earlier version of TF, it will be either slow or wrong.
# Faster-RCNN on COCO # Faster-RCNN on COCO
This example aims to provide a minimal (<1000 lines) multi-GPU implementation of ResNet50-Faster-RCNN on COCO. This example aims to provide a minimal (1.2k lines) multi-GPU implementation of ResNet50-Faster-RCNN on COCO.
## Dependencies ## Dependencies
+ TensorFlow > 1.4.0 (use tf-nightly-gpu for now) + TensorFlow >= 1.4.0
+ Install [pycocotools](https://github.com/pdollar/coco/tree/master/PythonAPI/pycocotools), OpenCV. + Install [pycocotools](https://github.com/pdollar/coco/tree/master/PythonAPI/pycocotools), OpenCV.
+ Pre-trained [ResNet50 model](https://goo.gl/6XjK9V) from tensorpack model zoo. + Pre-trained [ResNet50 model](https://goo.gl/6XjK9V) from tensorpack model zoo.
+ COCO data. It assumes the following directory structure: + COCO data. It assumes the following directory structure:
......
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