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)

相关推荐
GoCodingInMyWay6 分钟前
YOLO-Master 与 YOLO26 开始
yolo
懷淰メ6 小时前
python3GUI---基于PyQt5+YOLOv8+DeepSort的智慧行车可视化系统(详细介绍)
开发语言·yolo·计算机视觉·pyqt·yolov8·deepsort·车距
JicasdC123asd6 小时前
P6超大目标检测层改进YOLOv26四尺度特征金字塔与大感受野建模双重突破
yolo·目标检测·目标跟踪
JicasdC123asd6 小时前
PPHGNetV2高性能块改进YOLOv26密集连接与压缩激励双重突破
yolo
夏雨不在低喃1 天前
osnet模型和yolo模型的微调:冻结训练
yolo
向哆哆1 天前
人脸眼部特征检测数据集(千张图片已划分、已标注)适用于YOLO系列深度学习分类检测任务
深度学习·yolo·分类
Dev7z1 天前
基于YOLOv8面向家居场景的火焰烟雾图像识别系统
人工智能·yolo
童话名剑1 天前
YOLO v6(学习笔记)
yolo·目标检测·yolov6
前网易架构师-高司机2 天前
带标注的瓶盖识别数据集,识别率99.5%,可识别瓶盖,支持yolo,coco json,pascal voc xml格式
人工智能·yolo·数据集·瓶盖
一勺汤2 天前
YOLO26 改进、魔改| 部分通道注意力模块PAT,以轻量化并行结构融合局部卷积与增强型通道注意力,提升小目标、遮挡目标的检测效果。
yolo·注意力机制·轻量化·小目标·yolo26·yolo26改进·复杂场景