自学内容网 自学内容网

YOLOv5、v7、v8如何修改检测框文字颜色和大小

YOLOv5和YOLOv8默认的标签文字颜色为白色,但是在亮度较大的图片中文字不明显,就需要对标签文字的颜色进行修改

一、YOLOv5

打开X:\Anaconda\envs\your-env\Lib\site-packages\ultralytics\utils\plotting.py

X代表你的anaconda安装的盘,your-env代表你的YOLOv5环境名称

找到box_label这个用法,将原有的txt_color=(255, 255, 255)修改为txt_color=(0, 0, 0),标签文字颜色即可变为黑色

二、 YOLOv7

打开utils\plots.py,找到plot_one_box用法,修改cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 10, [0, 0, 0], thickness=22, lineType=cv2.LINE_AA),其中[0, 0, 0]代表BGR的黑色,thickness=22代表检测框的粗细

三、YOLOv8

打开ultralytics\utils\plotting.py文件,找到box_label这个用法,将原有的txt_color=(255, 255, 255)修改为txt_color=(0, 0, 0),标签文字颜色即可变为黑色


原文地址:https://blog.csdn.net/Yayisheng/article/details/140258796

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!