Add the switch of whether to display the result.

open-mmlab · Oct 3, 2022 · 01ba6a3 · 01ba6a3
1 parent 0a1c1c9
commit 01ba6a3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/demo/image_demo.py b/demo/image_demo.py
@@ -16,6 +16,8 @@ def parse_args():
     parser.add_argument('--out-file', default=None, help='Path to output file')
     parser.add_argument(
         '--device', default='cuda:0', help='Device used for inference')
+    parser.add_argument(
+        '--show', action='store_true', help='Show the detection results')
     parser.add_argument(
         '--palette',
         default='coco',
@@ -50,7 +52,7 @@ def main(args):
         img,
         data_sample=result,
         draw_gt=False,
-        show=True,
+        show=args.show,
         wait_time=0,
         out_file=args.out_file,
         pred_score_thr=args.score_thr)