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.