【发布时间】:2022-06-24 15:31:58
【问题描述】:
我正在使用 YOLOv5s 对自定义数据集进行对象检测,给定视频中有多个对象,有时标签文本和边界框厚度看起来很糟糕。如何自定义这些东西?
【问题讨论】:
标签: computer-vision object-detection yolov5
我正在使用 YOLOv5s 对自定义数据集进行对象检测,给定视频中有多个对象,有时标签文本和边界框厚度看起来很糟糕。如何自定义这些东西?
【问题讨论】:
标签: computer-vision object-detection yolov5
使用detect.py时,传入以下参数来调整标签和边界框:
--line-thickness 1
--hide-labels True
--hide-conf True
对于--line-thickness 参数,传入一个整数值来调整粗细,对于标签和置信度,它们默认设置为False。将它们设置为 True 将隐藏它们。
【讨论】:
小心使用--hide-labels True 或hide-conf True。
在某些情况下,您可能会遇到错误消息。只有 --hide-labels 和 --hide-conf 应该可以正常工作。
这是您可以与detect.py 一起使用的参数列表。
detect.py [-h] [--weights WEIGHTS [WEIGHTS ...]] [--source SOURCE]
[--data DATA] [--imgsz IMGSZ [IMGSZ ...]]
[--conf-thres CONF_THRES] [--iou-thres IOU_THRES]
[--max-det MAX_DET] [--device DEVICE] [--view-img]
[--save-txt] [--save-conf] [--save-crop] [--nosave]
[--classes CLASSES [CLASSES ...]] [--agnostic-nms]
[--augment] [--visualize] [--update] [--project PROJECT]
[--name NAME] [--exist-ok] [--line-thickness LINE_THICKNESS]
[--hide-labels] [--hide-conf] [--half] [--dnn]
【讨论】: