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)

相关推荐
长江后浪博客6 小时前
YOLO目标检测模型调参入门:从训练参数到检测效果优化
yolo·目标检测·机器学习·yolov8·模型调参
策案案案7 小时前
YOLO: 将AI Agents嵌入到IntelliJ IDEA
java·大数据·人工智能·笔记·yolo·microsoft·intellij-idea
公爵爱学习7 小时前
无人机视觉识别前序-Linux-YOLO安装
python·yolo·计算机视觉·conda·无人机
长江后浪博客10 小时前
Python + YOLOv8 疲劳驾驶 AI 视觉检测入门:从模型训练到 ONNX 实时摄像头检测完整实战
人工智能·python·yolo·疲劳驾驶检测·onnx·yolov8
JarmanYuo12 小时前
YOLO 涨点研究(九):多模态融合检测篇——RGB-红外融合的网络结构与论文全解析
yolo
YOLO数据集集合15 小时前
无花果目标检测数据集 |无花果检测 智慧农业 果实识别 目标检测 YOLO格式 深度学习数据集 计算机视觉9049期
yolo·目标检测·计算机视觉·农业·水果识别·无花果识别·无花果检测
YOLO数据集集合15 小时前
遥感影像树木检测数据集 | 遥感树木 目标检测 城市绿化 森林监测 YOLO格式9052期
人工智能·yolo·目标检测·计算机视觉·目标跟踪·树木检测·遥感树影
广州灵眸科技有限公司18 小时前
灵眸科技EAI3572-Core-L核心板即将发布!八核+4TOPS NPU,面向工业与边缘AI
linux·运维·服务器·数据库·yolo
Ysn07191 天前
YOLO-Master工程:modules.py 全景详解与 MoE 迁移启发
yolo
长江后浪博客2 天前
YOLOv26+500万彩色工业相机实现药品铝塑板颗粒完整度检测:缺粒识别、颜色检测与穹顶漫射光机器视觉方案
数码相机·yolo·机器视觉·yolov26·药品包装检测·铝塑板检测·彩色工业相机