我在使用yolov8进行检测时,想要在此基础上重新标注数据数据时发现了这个label文件中多了一列,是自己生成的置信度分数,解决方法只需要在检测的时候设置一个save_conf参数即可实现,如下:
from ultralytics import YOLO
model = YOLO("训练好的权重文件")
model.predict(source="图片文件", save=True, save_conf=False, save_txt=True, name='output', hide_conf=True)
# source后为要预测的图片数据集的的路径
# save=True为保存预测结果
# save_conf=False为不保存txt文件中的置信度分数
# save_txt=True为保存txt结果,但是yolov8本身当图片中预测不到异物时,不产生txt文件