Search code examples
computer-visionobject-detectionyolov5

How to change the bounding box thickness and label text in yolov5?

I am using YOLOv5s for object detection on custom datasets, there are multiple objects in given video, sometimes label text and bounding box thickness looks very bad. how can I customize these things?

Solution

  • when using detect.py, pass in the following arguments to adjust the labels and bounding boxes:

    --line-thickness 1
--hide-labels True
--hide-conf True

    For the --line-thickness argument, pass in an integer value to adjust the thickness, for labels and confidence, they are set to False by default. Setting them to True will hide them.