YOLOv8将输出结果3位有效数字改为4位有效数字

最新推荐文章于 2024-08-18 12:11:57 发布

Zack_Li_CS

最新推荐文章于 2024-08-18 12:11:57 发布

阅读量735

点赞数 3

文章标签： YOLO

本文链接：https://blog.csdn.net/m0_60763988/article/details/135168381

版权

文章介绍了如何在UltralyticsYOLOv8.0.202模型的val.py文件中，修改`print_results`函数的打印格式，从保留小数点后3位改为4位，以获取更精确的训练/验证集类别指标。同时，警告如果没有标签，将无法计算指标。

摘要由CSDN通过智能技术生成

1.采用的版本是Ultralytics YOLOv8.0.202

找到ultralytics/models/yolo/detect/val.py这个文件

2.找到print_results函数

    def print_results(self):
        """Prints training/validation set metrics per class."""
        pf = '%22s' + '%11i' * 2 + '%11.3g' * len(self.metrics.keys)  # print format
        LOGGER.info(pf % ('all', self.seen, self.nt_per_class.sum(), *self.metrics.mean_results()))
        if self.nt_per_class.sum() == 0:
            LOGGER.warning(
                f'WARNING ⚠️ no labels found in {self.args.task} set, can not compute metrics without labels')

3.修改pf = '%22s' + '%11i' * 2 + '%11.3g' * len(self.metrics.keys) 为 pf = '%22s' + '%11i' * 2 + '%11.4g' * len(self.metrics.keys)


    def print_results(self):
        """Prints training/validation set metrics per class."""
        pf = '%22s' + '%11i' * 2 + '%11.4g' * len(self.metrics.keys)  # print format
        LOGGER.info(pf % ('all', self.seen, self.nt_per_class.sum(), *self.metrics.mean_results()))
        if self.nt_per_class.sum() == 0:
            LOGGER.warning(
                f'WARNING ⚠️ no labels found in {self.args.task} set, can not compute metrics without labels')

修改成功。