Skip to content

Commit 0b172c5

Browse files
committed
updated confidence default
1 parent 656d75a commit 0b172c5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

partinet/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def step2(**params):
140140
@click.option('--source', type=str, default='inference/images', help='source', show_default=True) # file/folder, 0 for webcam
141141
@click.option('--num-classes', type=int, default=1, help='number of classes', show_default=True)
142142
@click.option('--img-size', type=int, default=1280, help='inference size (pixels)', show_default=True)
143-
@click.option('--conf-thres', type=float, default=0.0, help='object confidence threshold', show_default=True)
143+
@click.option('--conf-thres', type=float, default=0.1, help='object confidence threshold', show_default=True)
144144
@click.option('--iou-thres', type=float, default=0.2, help='IOU threshold for NMS', show_default=True)
145145
@click.option('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu', show_default=True)
146146
@click.option('--view-img', is_flag=True, help='display results')

0 commit comments

Comments
 (0)