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

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]

示例如下:

 

相关文章:

  • AI编程案例拆解|基于机器学习XX评分系统-后端篇
  • 深入理解算力:从普通电脑到宏观计算世界
  • 【Docker项目实战】使用Docker部署Caddy+vaultwarden密码管理工具(详细教程)
  • 如何在项目中使用双token机制?
  • 代码随想录算法训练营Day36
  • MyBatis XML 配置完整示例(含所有核心配置项)
  • 单片机-89C51部分:4、固件烧录
  • MAVLink协议:原理、应用与实践
  • Pytorch(无CPU搭建)+Jupyter
  • 代码随想录算法训练营第二十八天
  • Pygame入门:零基础打造你的第一个游戏窗口
  • 二、UI自动化测试02--元素定位方法
  • Leetcode刷题 由浅入深之哈希法——202. 快乐数
  • Adruino:传感器及步进电机
  • 我们分析前端生活。
  • 驱动开发硬核特训 · Day 21(上篇加强版):深入理解子系统机制与实战初探
  • 微服务架构下 MySQL 大表分库分表方案
  • 【Linux网络】构建与优化HTTP请求处理 - HttpRequest从理解到实现
  • std::mutex底层实现原理
  • Spring Boot集成RocketMQ
  • 持续更新丨伊朗内政部长:港口爆炸已致8人死亡750人受伤
  • 委员呼吁提高政府机构电话号码准确性,辽宁阜新回应
  • 云南鲁甸县一河滩突然涨水致4死,有人在救人过程中遇难
  • 一季度公募管理规模出炉:44家实现增长,4家规模环比翻倍
  • 传媒湃︱《金陵晚报》副刊“雨花石”5月起改为免费刊登
  • 体坛联播|卡马文加预计伤缺三个月,阿尔卡拉斯因伤退赛