使用pycocotools打印更多数据，并且数值扩展到小数点后6位（注意，修改后最好再还原！最好是一次性使用！也可以长期使用，只要记得这里改了的就行[手动狗头]）

孟孟单单

已于 2024-01-16 21:09:23 修改

阅读量687

点赞数 10

分类专栏：深度学习项目经验tips 文章标签： json YOLO

于 2024-01-16 18:56:50 首次发布

本文链接：https://blog.csdn.net/lwd19981223/article/details/135632159

版权

深度学习项目经验tips 专栏收录该内容

40 篇文章 11 订阅

订阅专栏

本文介绍了如何修改pycocotools库的代码，以打印出YOLOv7-tiny模型在小、中、大尺寸目标上的IoU=0.5的AP和AR值，同时提供了在本地和云服务器上进行代码调整的详细步骤。

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

1 写在前面

夹带私货，可能有用呢：YOLOv7-tiny，通过pycocotools包得到预测大中小尺寸目标的指标值

仅供参考！写这个代码的目的是能够打印出iou=0.50的AP、AR的小中大3个尺寸的值，原始代码是不会打印出来的！

最好是一次性使用！因为这个是在动pycocotools这个包的源代码了，所以最好是用完之后就当场改回去，以免影响其他模型使用pycocotools包！！！

2 代码效果

原始效果：

在这里插入图片描述

代码修改之后的效果：

在这里插入图片描述

3 代码修改

本地电脑上：修改F:\SoftWarelnstall\Anaconda3\envs\torch1.10\Lib\site- packages\pycocotools\cocoeval.py这个文件，别找错了虚拟环境了哟，我这里修改的是自己配置好的torch1.10虚拟环境里面的文件

AutoDL云服务器上：在./miniconda3/envs/py38/lib/python3.8/site-packages\pycocotools\cocoeval.py修改，在云服务器上我的虚拟环境名是py38

打印小数点后五位数：在cocoeval.py中定位到iStr =然后将后面的{:0.3f}里的3改为想要的小数点位数，例如我想要5位数就改为5

在这里插入图片描述

打印更多数据：在cocoeval.py中定位到def _summarizeDets():，然后进行如下修改👇

在这里插入图片描述

            # ------------------------------------------------------lwd edit-------------------------------------------------- #
            print('-' * 30)
            stats[12] = _summarize(1, iouThr=.5, areaRng='small', maxDets=self.params.maxDets[2])
            stats[13] = _summarize(1, iouThr=.5, areaRng='medium', maxDets=self.params.maxDets[2])
            stats[14] = _summarize(1, iouThr=.5, areaRng='large', maxDets=self.params.maxDets[2])
            print('-' * 30)
            stats[15] = _summarize(0, iouThr=.5, areaRng='small', maxDets=self.params.maxDets[2])
            stats[16] = _summarize(0, iouThr=.5, areaRng='medium', maxDets=self.params.maxDets[2])
            stats[17] = _summarize(0, iouThr=.5, areaRng='large', maxDets=self.params.maxDets[2])
            # ------------------------------------------------------lwd edit-------------------------------------------------- #