|
|
@@ -3,7 +3,7 @@
|
|
|
## 目录
|
|
|
|
|
|
* [paddlex.det.visualize](#1)
|
|
|
-* [paddlex.det.fraw_pr_curve](#2)
|
|
|
+* [paddlex.det.draw_pr_curve](#2)
|
|
|
* [paddlex.det.coco_error_analysis](#3)
|
|
|
* [paddlex.seg.visualize](#4)
|
|
|
* [paddlex.visualize_det](#5)
|
|
|
@@ -58,12 +58,12 @@ paddlex.det.draw_pr_curve(eval_details_file=None, gt=None, pred_bbox=None, pred_
|
|
|
**注意：**`eval_details_file`的优先级更高，只要`eval_details_file`不为None，就会从`eval_details_file`提取真值信息和预测结果做分析。当`eval_details_file`为None时，则用`gt`、`pred_mask`、`pred_mask`做分析。
|
|
|
|
|
|
### 使用示例
|
|
|
-点击下载如下示例中的[模型](https://bj.bcebos.com/paddlex/models/insect_epoch_270.zip)和[数据集](https://bj.bcebos.com/paddlex/datasets/insect_det.tar.gz)
|
|
|
+点击下载如下示例中的[模型](https://bj.bcebos.com/paddlex/2.0/faster_rcnn_e12.tar.gz)和[数据集](https://bj.bcebos.com/paddlex/datasets/insect_det.tar.gz)
|
|
|
|
|
|
> 方式一：分析训练过程中保存的模型文件夹中的评估结果文件`eval_details.json`，例如[模型](https://bj.bcebos.com/paddlex/models/insect_epoch_270.zip)中的`eval_details.json`。
|
|
|
```
|
|
|
import paddlex as pdx
|
|
|
-eval_details_file = 'insect_epoch_270/eval_details.json'
|
|
|
+eval_details_file = 'faster_rcnn_e12/eval_details.json'
|
|
|
pdx.det.draw_pr_curve(eval_details_file, save_dir='./insect')
|
|
|
```
|
|
|
> 方式二：分析模型评估函数返回的评估结果。
|
|
|
@@ -72,13 +72,13 @@ pdx.det.draw_pr_curve(eval_details_file, save_dir='./insect')
|
|
|
import paddlex as pdx
|
|
|
from paddlex import transforms as T
|
|
|
|
|
|
-model = pdx.load_model('insect_epoch_270')
|
|
|
+model = pdx.load_model('faster_rcnn_e12')
|
|
|
eval_dataset = pdx.datasets.VOCDetection(
|
|
|
data_dir='insect_det',
|
|
|
file_list='insect_det/val_list.txt',
|
|
|
label_list='insect_det/labels.txt',
|
|
|
transforms=model.test_transforms)
|
|
|
-metrics, evaluate_details = model.evaluate(eval_dataset, batch_size=8, return_details=True)
|
|
|
+metrics, evaluate_details = model.evaluate(eval_dataset, batch_size=1, return_details=True)
|
|
|
gt = evaluate_details['gt']
|
|
|
bbox = evaluate_details['bbox']
|
|
|
pdx.det.draw_pr_curve(gt=gt, pred_bbox=bbox, save_dir='./insect')
|
will-jl944