(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."""
        #------------------代码添加---------------------------------
        LOGGER.info(("%22s" + "%11s" * 6) % ("Class", "Images", "Instances", "Box(P", "R", "mAP50", "mAP50-95)"))
        #--------------------代码添加-------------------------------
        return ("%22s" + "%11s" * 6) % ("Class", "Images", "Instances", "Box(P", "R", "mAP50", "mAP50-95)")

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐