如图所示,训练完我发现,我的两类都是红色只不过一个深一点,一个浅一点,检测框也是这样,很容易混淆,请问怎么改变不同类是不同的颜色呢,在哪里修改呢,我用的yolov5-mater代码
yolov5检测框怎么定义颜色
如图所示,训练完我发现,我的两类都是红色只不过一个深一点,一个浅一点,检测框也是这样,很容易混淆,请问怎么改变不同类是不同的颜色呢,在哪里修改呢,我用的yolov5-mater代码
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
2条回答 默认 最新
- H3T 2023-03-27 13:02关注
在YoloV5检测框中定义不同的颜色可以在detect.py文件中进行修改。具体操作如下:
打开detect.py文件,找到plot_one_box函数,该函数用于绘制框框。
在plot_one_box函数中添加一个参数color=None,用于指定绘制框框时的颜色。
在plot_one_box函数中,使用color参数来指定绘制框框的颜色。可以根据类别不同来指定不同的颜色。
在detect.py文件中的plot_one_box函数中,可以根据类别来为每个类别指定颜色,如下所示:# 定义不同类别的颜色 colors = [[0, 255, 255], [255, 0, 0]] # 在绘制框框时,根据类别来指定颜色 color = colors[int(cls)]
其中,colors是一个颜色列表,列表中的每个元素对应一个类别的颜色,cls是检测框所属的类别,用于指定绘制框框时的颜色。
完成修改后,保存文件并重新运行检测脚本,即可看到不同类别的检测框颜色已经被修改了。
如果答案对您有所帮助,望采纳。本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥15 虚幻5 UE美术毛发渲染
- ¥15 CVRP 图论 物流运输优化
- ¥15 Tableau online 嵌入ppt失败
- ¥100 支付宝网页转账系统不识别账号
- ¥15 基于单片机的靶位控制系统
- ¥15 真我手机蓝牙传输进度消息被关闭了,怎么打开?(关键词-消息通知)
- ¥15 装 pytorch 的时候出了好多问题,遇到这种情况怎么处理?
- ¥20 IOS游览器某宝手机网页版自动立即购买JavaScript脚本
- ¥15 手机接入宽带网线,如何释放宽带全部速度
- ¥30 关于#r语言#的问题:如何对R语言中mfgarch包中构建的garch-midas模型进行样本内长期波动率预测和样本外长期波动率预测