Arain___的博客

原创 打印YOLOv5训练过程中每个类别的AP

YOLOv5训练状况下默认只显示mAP_0.5和mAP_0.5：0.95，想要查看最终每个类别的AP只能在训练结束后的PR曲线中查看。如果想在训练过程中打印，需要在val.py中270行附近修改。此时训练过程中每一轮都将显示每一个类别的AP。