diff --git a/partinet/__init__.py b/partinet/__init__.py index d8b9008..45701d6 100644 --- a/partinet/__init__.py +++ b/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() \ No newline at end of file + main()