Yolov5根据目标类别自定义识别框的颜色

修改下detect文件下的write results函数

# Write results
for *xyxy, conf, cls in reversed(det):
if save_txt:  # Write to filexywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()  # normalized xywhline = (cls, *xywh, conf) if opt.save_conf else (cls, *xywh)  # label formatwith open(txt_path + '.txt', 'a') as f:f.write(('%g ' * len(line)).rstrip() % line + '\n')if save_img or view_img:  # Add bbox to imagelabel = f'{names[int(cls)]} {conf:.2f}'#定义颜色color_dict = {'1': [0, 0, 250], '2': [190, 90, 92], '3': [142, 154, 78], '4': [2, 76, 82],'5': [119, 80, 5], '6': [189, 163, 234]}  # Rgb3个值：B G Rif names[int(cls)] == 'car':        #根据训练的文件.yaml中的name类中的名称修改color_single = color_dict['1']elif names[int(cls)] == 'person':        color_single = color_dict['2']plot_one_box(xyxy, im0, label=label, color=color_single, line_thickness=3)

参考文档：

(53条消息) 【目标检测】YOLOv5：标签中文显示/自定义颜色_yolov5中文标签_zstar-_的博客-CSDN博客

Yolov5根据目标类别自定义识别框的颜色相关推荐

Android自定义人脸识别框,android自定义虹软人脸识别框/人脸抓拍框/人脸追踪框...
有一两个月没有更新博客了,最近一直在忙人脸识别的项目,先将项目中更改虹软人脸框的方法抽取出来,希望对大家有所帮助. 尊重原创,转载请注明出处: http://blog..net/qq137722697 ...
android 人脸识别边框_android自定义Arcface人脸识别框/人脸抓拍框/人脸追踪框
为什么要改? 先来看看sdk demo中提供的人脸框样式,这个框看上去并不是非常美观(个人觉得) 在这里插入图片描述再看看下面这个框是不是就要顺眼一点在这里插入图片描述怎么换? 先来看看原始的画 ...
Yolov5 安装详细教程及目标检测和识别
文章内容: 1.在 Anaconda 环境下,进行目标检测程序(Yolov5)的下载及安装,实现 Yolov5 的整体安装: 2.通过调用手机摄像头进行简单的目标检测和识别. 目录 1 任务目标 2 ...
YOLO 目标检测识别框不显示文字标签（已解决）
问题描述类似这种报错: 会导致yolo识别框上没有文字标签: 报错原因文件夹data/labels/ 中有字母字符的图片.YOLO 使用它们在图片中绘制标签.如果没有该文件夹或者已删除该文件夹,因 ...
【目标检测】英雄联盟能用YOLOv5实时目标检测了支持onnx推理
目录一.项目介绍二.项目结构三.准备数据 1.数据标注 2.数据转换格式四.执行训练 1.anchors文件 2.标签文件 3.预训练模型 4.训练数据 5.修改配置 6.执行训练五.执行预 ...
基于YOLOv5的目标检测系统详解（附MATLAB GUI版代码）
摘要:本文重点介绍了基于YOLOv5目标检测系统的MATLAB实现,用于智能检测物体种类并记录和保存结果,对各种物体检测结果可视化,提高目标识别的便捷性和准确性.本文详细阐述了目标检测系统的原理,并给 ...
基于YOLOv5的中式快餐店菜品识别系统
基于YOLOv5的中式快餐店菜品识别系统[金鹰物联智慧食堂项目] 摘要本文基于YOLOv5v6.1提出了一套适用于中式快餐店的菜品识别自助支付系统,综述了食品识别领域的发展现状,简要介绍了YOLOv ...
从R-CNN到Faster-RCNN再到YOLOV5，目标检测网络发展概述
R-CNN R-CNN由Ross Girshick于2014年提出,R-CNN首先通过选择性搜索算法Selective Search从一组对象候选框中选择可能出现的对象框,然后将这些选择出来的对象框中 ...
基于YOLOv5的舰船检测与识别系统（Python+清新界面+数据集）
摘要:基于YOLOv5的舰船检测与识别系统用于识别包括渔船.游轮等多种海上船只类型,检测船舰目标并进行识别计数,以提供海洋船只的自动化监测和管理.本文详细介绍船舰类型识别系统,在介绍算法原理的同时,给 ...
pytorch深度学习-机器视觉-目标检测和边界框简介
机器视觉之目标检测和边界框简介在图像分类任务里,我们假设图像里只有一个主体目标,并关注如何识别该目标的类别.然而,很多时候图像里有多个我们感兴趣的目标,我们不仅想知道它们的类别,还想得到它们在图像中 ...

Yolov5根据目标类别自定义识别框的颜色

修改下detect文件下的write results函数

参考文档：

Yolov5根据目标类别自定义识别框的颜色相关推荐

最新文章

热门文章