Refine infer (#2635)

* refine README & use a better visualization for doc

* fix typo in GETTING STARTED

README.md

@@ -36,16 +36,16 @@ multi-GPU training.
 
 Supported Architectures:
 
-|                    | ResNet | ResNet-vd <sup>[1](#vd)</sup> | ResNeXt | SENet | MobileNet | DarkNet |
-|--------------------|:------:|------------------------------:|:-------:|:-----:|:---------:|:-------:|
-| Faster R-CNN       | ✓      |                             ✓ | ✓       | ✓     | ✗         | ✗       |
-| Faster R-CNN + FPN | ✓      |                             ✓ | ✓       | ✓     | ✗         | ✗       |
-| Mask R-CNN         | ✓      |                             ✓ | ✓       | ✓     | ✗         | ✗       |
-| Mask R-CNN + FPN   | ✓      |                             ✓ | ✓       | ✓     | ✗         | ✗       |
-| Cascade R-CNN      | ✓      |                             ✗ | ✗       | ✗     | ✗         | ✗       |
-| RetinaNet          | ✓      |                             ✗ | ✗       | ✗     | ✗         | ✗       |
-| Yolov3             | ✓      |                             ✗ | ✗       | ✗     | ✓         | ✓       |
-| SSD                | ✗      |                             ✗ | ✗       | ✗     | ✓         | ✗       |
+|                    | ResNet | ResNet-vd <sup>[1](#vd)</sup> | SENet | MobileNet | DarkNet |
+|--------------------|:------:|------------------------------:|:-----:|:---------:|:-------:|
+| Faster R-CNN       | ✓      |                             ✓ | ✓     | ✗         | ✗       |
+| Faster R-CNN + FPN | ✓      |                             ✓ | ✓     | ✗         | ✗       |
+| Mask R-CNN         | ✓      |                             ✓ | ✓     | ✗         | ✗       |
+| Mask R-CNN + FPN   | ✓      |                             ✓ | ✓     | ✗         | ✗       |
+| Cascade R-CNN      | ✓      |                             ✗ | ✗     | ✗         | ✗       |
+| RetinaNet          | ✓      |                             ✗ | ✗     | ✗         | ✗       |
+| Yolov3             | ✓      |                             ✗ | ✗     | ✓         | ✓       |
+| SSD                | ✗      |                             ✗ | ✗     | ✓         | ✗       |
 
 <a name="vd">[1]</a> [ResNet-vd](https://arxiv.org/pdf/1812.01187) models offer much improved accuracy with negligible performance cost.
 
@@ -75,8 +75,8 @@ be saved in `output`.
 ```bash
 export PYTHONPATH=`pwd`:$PYTHONPATH
 python tools/infer.py -c configs/mask_rcnn_r50_1x.yml \
-    -o weights=https://paddlemodels.bj.bcebos.com/object_detection/mask_rcnn_r50_1x.tar
-    -infer_img=demo/000000570688.jpg
+    -o weights=https://paddlemodels.bj.bcebos.com/object_detection/mask_rcnn_r50_1x.tar \
+    --infer_img=demo/000000570688.jpg
 ```
 
 For detailed training and evaluation workflow, please refer to [GETTING_STARTED.md](docs/GETTING_STARTED.md).

docs/GETTING_STARTED.md

@@ -20,7 +20,7 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml
 
 ```bash
 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
-python tools/train.py -c =configs/faster_rcnn_r50_1x.yml
+python tools/train.py -c configs/faster_rcnn_r50_1x.yml
 ```
 
 - Datasets is stored in `dataset/coco` by default (configurable).

tools/infer.py

@@ -166,7 +166,8 @@ def main():
             image_path = imid2path[int(im_id)]
             image = Image.open(image_path).convert('RGB')
             image = visualize_results(image,
-                                      int(im_id), catid2name, 0.5, bbox_results,
+                                      int(im_id), catid2name,
+                                      FLAGS.draw_threshold, bbox_results,
                                       mask_results, is_bbox_normalized)
             save_name = get_save_image_name(FLAGS.output_dir, image_path)
             logger.info("Detection bbox results save in {}".format(save_name))
@@ -190,5 +191,10 @@ if __name__ == '__main__':
         type=str,
         default="output",
         help="Directory for storing the output visualization files.")
+    parser.add_argument(
+        "--draw_threshold",
+        type=float,
+        default=0.5,
+        help="Threshold to reserve the result for visualization.")
     FLAGS = parser.parse_args()
     main()