rtdertv2有可视化pr曲线的代码吗

[xiaocongxin]

Star RTDETR
请先在RTDETR主页点击star以支持本项目
Star RTDETR to help more people discover this project.

---

Describe the bug
A clear and concise description of what the bug is.
If applicable, add screenshots to help explain your problem.

To Reproduce
Steps to reproduce the behavior.
请问rtdertv2有可视化pr曲线的代码吗

[lyuwenyu]

目前没有相关的代码的；

[xiaocongxin]

目前没有相关的代码的；

好的，谢谢。想要绘制混淆矩阵，能否指点一下可以修改哪部分代码

[lyuwenyu]

需要重写一下accumulate；这个函数内部有一个_summarize；可以看到self.eval['precision'], self.eval['recall']怎么使用；不过还是建议使用第三方库直接可以画的那种

RT-DETR/rtdetrv2_pytorch/src/data/dataset/coco_eval.py

Line 73 in 0b6972d

coco_eval.accumulate()

---

https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocotools/cocoeval.py#L411-L418