Merge pull request #35 from WEHI-ResearchComputing/MihinP-patch-1

Enable saving results and confidences by default

Author: MihinP

partinet/__init__.py

@@ -144,8 +144,8 @@ def step2(**params):
 @click.option('--iou-thres', type=float, default=0.2, help='IOU threshold for NMS', show_default=True)
 @click.option('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu', show_default=True)
 @click.option('--view-img', is_flag=True, help='display results')
-@click.option('--save-txt', is_flag=True, help='save results to *.txt')
-@click.option('--save-conf', is_flag=True, help='save confidences in --save-txt labels')
+@click.option('--save-txt', is_flag=True, default=True, help='save results to *.txt, (default: enabled) ')
+@click.option('--save-conf', is_flag=True, default=True, help='save confidences in --save-txt labels (default: enabled)')
 @click.option('--nosave', is_flag=True, help='do not save images/videos')
 @click.option('--classes', multiple=True, type=int, default=[], help='filter by class: --classes 0, or --classes 0 --classes 2 --classes 3', show_default=True)
 @click.option('--agnostic-nms', is_flag=True, help='class-agnostic NMS')
@@ -208,4 +208,4 @@ def test(**params):
         partinet.DynamicDet.test.main(opt)
 
 if __name__ == "__main__":
-    main()
+    main()