一般yolov8使用分类网络预测视频时左上角显示的分类结果是白色的,有的时候会看不清楚,因此需要更改显示颜色以更好的可视化。
# 使用YOLOv8进行预测
results = model(frame)
# 可视化结果
annotated_frame = results[0].plot()
cv2.imshow('Annotated Frame', annotated_frame)
这里不能添加确定color的参数,经过一路的debug,来到
/ultralytics/engine/results.py
文件下
第562行,这里的txt_color就是显示的文字颜色,原本是(255,255,255)白色
将其修改为你想要的颜色即可,我直接设置为了(0,0,0),再次运行就能看见左上角的预测结果变为了黑色。