未验证 提交 7ed2aa7d 编写于 作者: G Guanghua Yu 提交者: GitHub

Update README.md

上级 e263e885
...@@ -23,7 +23,7 @@ ...@@ -23,7 +23,7 @@
| 模型 | 策略 | mAP | TRT-FP32 | TRT-FP16 | TRT-INT8 | 配置文件 | 模型 | | 模型 | 策略 | mAP | TRT-FP32 | TRT-FP16 | TRT-INT8 | 配置文件 | 模型 |
| :-------- |:-------- |:--------: | :----------------: | :----------------: | :---------------: | :----------------------: | :---------------------: | | :-------- |:-------- |:--------: | :----------------: | :----------------: | :---------------: | :----------------------: | :---------------------: |
| PicoDet-S-NPU | Baseline | 30.1 | - | - | - | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/picodet/picodet_s_416_coco_npu.yml) | [Model](https://bj.bcebos.com/v1/paddle-slim-models/act/picodet_s_416_coco_npu.tar) | | PicoDet-S-NPU | Baseline | 30.1 | - | - | - | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/picodet/picodet_s_416_coco_npu.yml) | [Model](https://bj.bcebos.com/v1/paddle-slim-models/act/picodet_s_416_coco_npu.tar) |
| PicoDet-S-NPU | 量化训练 | 29.7 | - | - | - | [config](https://github.com/PaddlePaddle/PaddleSlim/tree/develop/demo/full_quantization/picodet/configs/picodet_npu_with_postprocess.yaml) | [Model](https://bj.bcebos.com/v1/paddle-slim-models/act/picodet_s_npu_quant.tar) | | PicoDet-S-NPU | 量化训练 | 29.7 | - | - | - | [config](https://github.com/PaddlePaddle/PaddleSlim/tree/develop/example/full_quantization/picodet/configs/picodet_npu_with_postprocess.yaml) | [Model](https://bj.bcebos.com/v1/paddle-slim-models/act/picodet_s_npu_quant.tar) |
- mAP的指标均在COCO val2017数据集中评测得到,IoU=0.5:0.95。 - mAP的指标均在COCO val2017数据集中评测得到,IoU=0.5:0.95。
...@@ -114,12 +114,19 @@ python run.py --config_path=./configs/picodet_npu.yaml --save_dir='./output/' ...@@ -114,12 +114,19 @@ python run.py --config_path=./configs/picodet_npu.yaml --save_dir='./output/'
#### 3.5 测试模型精度 #### 3.5 测试模型精度
使用eval.py脚本得到模型的mAP: - 使用eval.py脚本得到模型的mAP:
``` ```
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python eval.py --config_path=./configs/picodet_npu_with_postprocess.yaml python eval.py --config_path=./configs/picodet_npu_with_postprocess.yaml
``` ```
- 使用ONNXRuntime测试模型mAP:
```
python onnxruntime_eval.py \
--reader_config=configs/picodet_reader.yml \
--model_path=picodet_s_416_coco_npu.onnx
```
**注意** **注意**
- 要测试的模型路径可以在配置文件中`model_dir`字段下进行修改。 - 要测试的模型路径可以在配置文件中`model_dir`字段下进行修改。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册