yolov5绘制pr曲线

用的yolov5 5.0版本 训练完以后会有一个pr曲线图,一共5个类,但是自动生成的这个曲线中有几类目标的曲线不想要
请问怎么生成只想要的几个类的pr曲线

在val.py中搜索下ap_per_class跳转过去自己改下代码,哪些类不要去掉就是了