mmdet/apis/下没有evaluate.py文件
时间: 2023-06-29 10:21:00 浏览: 107
evaluate.py
非常抱歉,我再次犯了一个错误。在mmdetection 2.x版本中,`evaluate.py`文件已经被移动到了`mmdet/core/evaluation/`目录下。下面是正确的修改步骤:
1. 在训练配置文件中找到`evaluation`部分,将`metric`的值改为`mAP`,并添加以下两行代码:
```
iou_thr=0.75
classwise=True
```
这样就可以输出map75和class-wise AP。
2. 若要输出map50,需要在`mmdet/core/evaluation/eval_hooks.py`文件中进行修改。打开`eval_hooks.py`文件,找到`results = _accumulate_results()`这一行代码,然后在该行代码之前添加以下代码:
```
iou_thr = 0.5
coco_eval = results['coco_eval']
coco_eval.params.iou_thr = iou_thr
```
修改后的代码如下:
```
results = _accumulate_results()
iou_thr = 0.5
coco_eval = results['coco_eval']
coco_eval.params.iou_thr = iou_thr
```
3. 运行训练脚本,训练完成后会在`work_dirs`目录下生成一个`latest.pth`文件和一个`latest.json`文件。`latest.json`文件就是训练结果的json文件,其中包含了map、map75、map50和class-wise AP的值。
阅读全文