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)

相关推荐
YOLO数据集集合2 小时前
无花果目标检测数据集 |无花果检测 智慧农业 果实识别 目标检测 YOLO格式 深度学习数据集 计算机视觉9049期
yolo·目标检测·计算机视觉·农业·水果识别·无花果识别·无花果检测
YOLO数据集集合2 小时前
遥感影像树木检测数据集 | 遥感树木 目标检测 城市绿化 森林监测 YOLO格式9052期
人工智能·yolo·目标检测·计算机视觉·目标跟踪·树木检测·遥感树影
广州灵眸科技有限公司5 小时前
灵眸科技EAI3572-Core-L核心板即将发布!八核+4TOPS NPU,面向工业与边缘AI
linux·运维·服务器·数据库·yolo
Ysn071915 小时前
YOLO-Master工程:modules.py 全景详解与 MoE 迁移启发
yolo
长江后浪博客1 天前
YOLOv26+500万彩色工业相机实现药品铝塑板颗粒完整度检测:缺粒识别、颜色检测与穹顶漫射光机器视觉方案
数码相机·yolo·机器视觉·yolov26·药品包装检测·铝塑板检测·彩色工业相机
Ysn07192 天前
YOLO-Master工程:experts.py 专家结构详解与目标检测专家设计启发
人工智能·yolo·目标检测
YOLO视觉与编程2 天前
YOLO / Labelme目标分割数据集增强扩充软件v1.0.0适用于YOLO全版本
人工智能·深度学习·yolo·计算机视觉
JarmanYuo2 天前
YOLO 涨点研究(六):网络结构改进之小目标增强篇1——无人机视角下的车辆与行人检测
人工智能·pytorch·python·yolo·计算机视觉·无人机
hans汉斯2 天前
【计算机科学与应用】基于联合熵驱动改进麻雀搜索优化VMD的DAS信号去噪方法
深度学习·算法·yolo·软件工程·汉斯出版社
CoderIsArt3 天前
印花缺陷检测方案二C# + Halcon 引擎 + YOLO 融合实现
yolo