From 47940fd684f9387d4496d198a76a27a261fb9602 Mon Sep 17 00:00:00 2001 From: Mihin <116701142+MihinP@users.noreply.github.com> Date: Fri, 23 Jan 2026 10:47:13 +1100 Subject: [PATCH] Enable saving results and confidences by default --- partinet/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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()