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),标签文字颜色即可变为黑色