在 YOLOv7 中添加各个类别的 mAP 可以通过以下步骤实现:
1. 首先,需要在训练时设置 `--project` 和 `--name` 参数,例如:
```
python train.py --img 640 --batch 16 --epochs 100 --data coco.yaml --weights yolov5s.pt --project my_project --name my_model
```
2. 然后,在训练完成后,可以使用 `test.py` 脚本计算 mAP。需要设置 `--data`、`--weights`、`--project` 和 `--name` 参数,例如:
```
python test.py --img 640 --conf 0.001 --data coco.yaml --weights runs/my_project/my_model/weights/best.pt --project my_project --name my_model
```
3. 最后,可以在 `runs/my_project/my_model/results.txt` 文件中查看各个类别的 mAP 值。