yolov5 5.0隐藏标签

yanmengying

于 2023-06-09 10:51:58 发布

阅读量673

点赞数 1

文章标签： YOLO

本文链接：https://blog.youkuaiyun.com/yanmengying/article/details/131122731

版权

在Yolov55.0的detect.py脚本中，若要隐藏对象检测的标签，可以注释掉第112行附近的代码，删除与`names[int(cls)]`相关的内容，只保留物体的置信度(conf)。这将使得框出的物体不再显示类别名称，仅显示置信度分数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

yolov5 5.0中没有--hide-labels这个选项

在detect.py文件中，112行左右，把原来的label注释掉，在原来label的基础上去掉names的内容，就ok了

                    if save_img or view_img:  # Add bbox to image
                        # label = f'{names[int(cls)]} {conf:.2f}'
                        label = f'{conf:.2f}'
                        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

原来

之后