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)

相关推荐
weixin_156241575763 小时前
基于YOLOv8深度学习花卉识别系统摄像头实时图片文件夹多图片等另有其他的识别系统可二开
大数据·人工智能·python·深度学习·yolo
QQ676580083 小时前
AI赋能轨道交通智能巡检 轨道交通故障检测 轨道缺陷断裂检测 轨道裂纹识别 鱼尾板故障识别 轨道巡检缺陷数据集深度学习yolo第10303期
人工智能·深度学习·yolo·智能巡检·轨道交通故障检测·鱼尾板故障识别·轨道缺陷断裂检测
云程笔记3 小时前
002.计算机视觉与目标检测发展简史:从传统方法到深度学习
深度学习·yolo·目标检测·计算机视觉
weixin_156241575763 小时前
基于YOLO深度学习的动物检测与识别系统
人工智能·深度学习·yolo
赵药师7 小时前
YOLO中task.py改复杂的模块
python·深度学习·yolo
小高求学之路7 小时前
本地MP4视频文件转为视频流
yolo·计算机视觉
QQ676580089 小时前
城市治理之河道污染识别 无人机河道污染巡检 塑料带识别 瓶子图像识别 深度学习垃圾识别第10384期
人工智能·深度学习·yolo·河道污染·无人机河道污染·瓶子图像·塑料袋识别
一勺汤10 小时前
双骨干并行,极简融合:一款高效双 Backbone 目标检测网络详解(单模态增强创新点,双模态检测)
yolo·多模态·双backbone·yolo双backbone·单模态·yolo26双backbone·yolo11双backbone
weixin_1562415757612 小时前
基于YOLO深度学习的运动品牌检测与识别系统
人工智能·深度学习·yolo·识别·模型、
爱喝雪碧的可乐12 小时前
[特殊字符] YOLO26 实战教程:从 0 到 1 完成自定义数据集训练全流程 | 附性能对比 + YOLOv5 迁移指南
yolo