YOLOv5指定标签框背景颜色和标签字

(个人实验修改记录备份,非教程)

修改文件为库函数里面的plotting.py,路径在ultralytics/utils/plotting.py,前面的路径视安装环境的位置而异。不过这个路径并不重要,在detect.py里面找到Annotation类直接跳转即可

标签相关的绘制在 box_label 函数中实现(line277-334)

line324 处的 color 和 line331 处的 txt_color 改成指定颜色

python 复制代码
cv2.rectangle(self.im, p1, p2, color, -1, cv2.LINE_AA)  # filled  color->(255,255,255)
cv2.putText(
    self.im,
    label,
    (p1[0], p1[1] - 2 if outside else p1[1] + h - 1),
    0,
    self.sf,
    txt_color,    # txt_color->(84, 74, 98)
    thickness=self.tf,
    lineType=cv2.LINE_AA,
)

不显示score只需在detect时在指定后加上 --hide-conf

detect指定记录: python detect.py --weights /data/zy/code/yolov5/runs/train/exp7/weights/best.pt --source /data/zy/dataset/cooper001/newsplit/yolo/images/test/ --line-thickness 1 --hide-conf

(11,code/yolov5)

相关推荐
Coding茶水间1 小时前
基于深度学习的面部口罩检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
musk12121 小时前
YOLOv8n模型微调全指南:从环境搭建到技能储备 (内容由 AI 生成)
人工智能·yolo
懷淰メ2 小时前
【AI加持】基于PyQt5+YOLOv8+DeepSeek的太阳能电池板缺陷检测系统(详细介绍)
yolo·目标检测·计算机视觉·pyqt5·检测系统·deepseek·太阳能电池
lxmyzzs2 小时前
【图像算法 - 36】医疗应用:基于 YOLOv12 与 OpenCV 的高精度脑肿瘤检测系统实现
python·深度学习·opencv·yolo·计算机视觉·脑肿瘤检测
boligongzhu19 小时前
ubuntu20.04搭建YOLOv11 GPU运行环境
linux·yolo·ubuntu·机器人
self-motivation20 小时前
征机器人领域主流模型量化,评测,优化,部署工具model_optimizer的开源合作开发
yolo·机器人·量化·foundationpose·pi0.5
Coding茶水间21 小时前
基于深度学习的火焰检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
paopao_wu21 小时前
人脸检测与识别-InsightFace:向量相似性搜索Faiss
人工智能·yolo·目标检测·ocr·faiss
a1111111111ss1 天前
KANConv
yolo
szxinmai主板定制专家1 天前
基于x86和ARM的EtherCAT运动控制器,最大支持32轴,支持codesys和实时系统优化
arm开发·人工智能·嵌入式硬件·yolo