Skip to content

Commit

Permalink
Add detect.py --hide-conf --hide-labels --line-thickness options (ult…
Browse files Browse the repository at this point in the history
…ralytics#2658)

* command line option for line thickness and hiding labels

* command line option for line thickness and hiding labels

* command line option for line thickness and hiding labels

* command line option for line thickness and hiding labels

* command line option for line thickness and hiding labels

* command line option for hiding confidence values

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
(cherry picked from commit f662c18)
  • Loading branch information
Ashafix authored and Lechtr committed Apr 25, 2021
1 parent b55fe25 commit 185496c
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions detect.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,8 +110,9 @@ def detect(opt):

if save_img or opt.save_crop or view_img: # Add bbox to image
c = int(cls) # integer class
label = f'{names[c]} {conf:.2f}'
plot_one_box(xyxy, im0, label=label, color=colors[c], line_thickness=3)
label = None if opt.hide_labels else (names[c] if opt.hide_conf else f'{names[c]} {conf:.2f}')

plot_one_box(xyxy, im0, label=label, color=colors[c], line_thickness=opt.line_thickness)
if opt.save_crop:
save_one_box(xyxy, im0s, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)

Expand Down Expand Up @@ -169,6 +170,9 @@ def detect(opt):
parser.add_argument('--project', default='runs/detect', help='save results to project/name')
parser.add_argument('--name', default='exp', help='save results to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
parser.add_argument('--line-thickness', default=3, type=int, help='bounding box thickness (pixels)')
parser.add_argument('--hide-labels', default=True, action='store_true', help='hide labels')
parser.add_argument('--hide-conf', default=True, action='store_true', help='hide confidences')
opt = parser.parse_args()
print(opt)
check_requirements(exclude=('pycocotools', 'thop'))
Expand Down

0 comments on commit 185496c

Please sign in to comment.