当前位置: 首页 > web >正文

YOLOv5修改检测框颜色,粗细,标签大小,标签名称

一、修改检测框颜色,粗细,标签大小

在detect.py中找到

ctrl+点击Annotator即可自动跳转到Annotator类所在位置。

如果没有自动跳转,可通过以下路径查找:

anaconda3/envs/yolo(自己创建的虚拟环境名字)/lib/python3.8/site-packages/ultralytics/utils/plotting.py

将对应的color、thickness、fontScale修改为自己需要的格式即可。

我这里作出如下修改:

修改前结果:

a2c7ac5aa1294ba5a131183966dc9f50.png 

修改后结果: 

二、修改标签名 

尝试在coco.yaml文件中修改,发现没有用。因为detect时使用的模型已经在训练阶段就将各个标签都写入模型里面了。

可通过直接覆盖model.names属性解决。

在detect.py中找到加载模型的代码。

在代码

model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half)

 stride, names, pt = model.stride, model.names, model.pt

之间插入以下代码即可覆盖之前的标签。可将其改为中文,注意类别继续要保持80,不然会报错。

model.names = ["person",        # 0"bicycle",       # 1"car",           # 2"motorcycle",    # 3"airplane",      # 4"bus",           # 5"train",         # 6"truck",         # 7"boat",          # 8"traffic light", # 9 "fire hydrant",  # 10 "stop sign",     # 11 "parking meter", # 12 "bench",         # 13"bird",          # 14"cat",           # 15"dog",           # 16"horse",         # 17"sheep",         # 18"cow",           # 19"elephant",      # 20"bear",          # 21"zebra",         # 22"giraffe",       # 23"backpack",      # 24"umbrella",      # 25"handbag",       # 26"tie",           # 27"suitcase",      # 28"frisbee",       # 29"skis",          # 30"snowboard",     # 31"sports ball",   # 32 "kite",          # 33"baseball bat",  # 34 "baseball glove",# 35 "skateboard",    # 36"surfboard",     # 37"tennis racket", # 38 "bottle",        # 39"wine glass",    # 40 "cup",           # 41"fork",          # 42"knife",         # 43"spoon",         # 44"bowl",          # 45"banana",        # 46"apple",         # 47"sandwich",      # 48"orange",        # 49"broccoli",      # 50"carrot",        # 51"hot dog",       # 52 "pizza",         # 53"donut",         # 54"cake",          # 55"chair",         # 56"couch",         # 57"potted plant",  # 58 "bed",           # 59"dining table",  # 60 "toilet",        # 61"tvmonitor",     # 62 "laptop",        # 63"mouse",         # 64"remote",        # 65"keyboard",      # 66"cell phone",    # 67 "microwave",     # 68"oven",          # 69"toaster",       # 70"sink",          # 71"refrigerator",  # 72"book",          # 73"clock",         # 74"vase",          # 75"scissors",      # 76"teddy bear",    # 77 "hair dryer",    # 78 "toothbrush"     # 79]

示例如下:

 

http://www.xdnf.cn/news/2221.html

相关文章:

  • 提示词的神奇魔力——如何通过它改变AI的输出
  • 7.Geometric Intersection: Interval
  • [实战] 卡尔曼滤波:原理、推导与卫星导航应用仿真(完整代码)
  • 若干查找算法
  • Vue3 组件通信与插槽
  • 未雨绸缪:应对软件开发变更的生存之道
  • 23种设计模式-行为型模式之观察者模式(Java版本)
  • 理想星环OS选择NuttX作为MCU侧OS的核心原因分析​
  • 树莓派学习专题<9>:使用V4L2驱动获取摄像头数据--设定分辨率和帧率
  • ASP.NET CORE部署IIS的三种方式
  • 第14节:传统图像特征提取 - 形状特征(HOG、SIFT与SURF)
  • 【fork初体验】
  • 数据结构手撕--【堆】
  • 【LeetCode】11.盛最多水的容器
  • 系列位置效应——AI与思维模型【80】
  • 鸿蒙代码@Builder
  • 关于调度策略的系统性解析与物流机器人应用实践
  • Universal Value Function Approximators 论文阅读(强化学习,迁移?)
  • 介绍常用的退烧与消炎药
  • 【Flume 】Windows安装步骤、配置环境
  • Llama factory如何全参数微调 Qwen2.5-7B-Instruct 模型并导入Ollama推理(详细版)
  • 大一下第一次考核题解
  • Linux文件目录操作实战
  • 【C++】15. 模板进阶
  • 【含文档+PPT+源码】基于Python的美食数据的设计与实现
  • llama factory 命令行推理流程
  • MUC基本知识
  • 电子电器架构 --- 乘用车电气/电子架构开发的关键挑战与应对策略
  • Shell编程之正则表达式
  • c++弹窗