(1)从官网下载的 yolov8,进行 val时,只输出batch 0 1 2 3 。其他的 预测图片不输出。
想要输出全部的val图片,请进行以下修改:
/ultralytics-8.1.0/ultralytics/engine/validator.py 第187行
#if self.args.plots and batch_i < 3: #将batch_i<3 去掉
#如下:
if self.args.plots:
self.plot_val_samples(batch, batch_i)
self.plot_predictions(batch, preds, batch_i)
源码中,batch_i<3 去掉,然后所有的预测图像就可以全部输出了。
另外,val.py中,batch ,可以设置为16,然后得到的预测图片中,是4*4。
(2val.py输出 PR曲线
yolov8源码中,val.py运行后,只会得到jpg图片,并没有PR曲线等。
如果想获取PR曲线等,按照下面修改:
/ultralytics-8.1.0/ultralytics/models/yolo/detect/val.py
大约在80行,增加代码:
def get_desc(self):
"""Return a formatted string summarizing class metrics of YOLO model."""