|
|
@@ -12,6 +12,10 @@ |
|
|
- [Script Parameters](#script-parameters) |
|
|
- [Script Parameters](#script-parameters) |
|
|
- [Training Process](#training-process) |
|
|
- [Training Process](#training-process) |
|
|
- [Evaluation Process](#evaluation-process) |
|
|
- [Evaluation Process](#evaluation-process) |
|
|
|
|
|
- [Inference Process](#inference-process) |
|
|
|
|
|
- [Export MindIR](#export-mindir) |
|
|
|
|
|
- [Infer on Ascend310](#infer-on-ascend310) |
|
|
|
|
|
- [result](#result) |
|
|
- [Model Description](#model-description) |
|
|
- [Model Description](#model-description) |
|
|
- [Performance](#performance) |
|
|
- [Performance](#performance) |
|
|
- [Evaluation Performance](#evaluation-performance) |
|
|
- [Evaluation Performance](#evaluation-performance) |
|
|
@@ -479,6 +483,37 @@ result: {'top_5_accuracy': 0.9342589628681178, 'top_1_accuracy': 0.7680657810499 |
|
|
|
|
|
|
|
|
``` |
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
## Inference Process |
|
|
|
|
|
|
|
|
|
|
|
### [Export MindIR](#contents) |
|
|
|
|
|
|
|
|
|
|
|
```shell |
|
|
|
|
|
python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] |
|
|
|
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
The ckpt_file parameter is required, |
|
|
|
|
|
`EXPORT_FORMAT` should be in ["AIR", "MINDIR"] |
|
|
|
|
|
|
|
|
|
|
|
### Infer on Ascend310 |
|
|
|
|
|
|
|
|
|
|
|
Before performing inference, the mindir file must bu exported by `export.py` script. We only provide an example of inference using MINDIR model. |
|
|
|
|
|
Current batch_Size can only be set to 1. The precision calculation process needs about 70G+ memory space, otherwise the process will be killed for execeeding memory limits. |
|
|
|
|
|
|
|
|
|
|
|
```shell |
|
|
|
|
|
# Ascend310 inference |
|
|
|
|
|
bash run_infer_310.sh [MINDIR_PATH] [DATA_PATH] [DEVICE_ID] |
|
|
|
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
- `DEVICE_ID` is optional, default value is 0. |
|
|
|
|
|
|
|
|
|
|
|
### result |
|
|
|
|
|
|
|
|
|
|
|
Inference result is saved in current path, you can find result like this in acc.log file. |
|
|
|
|
|
|
|
|
|
|
|
```bash |
|
|
|
|
|
top1_accuracy:70.42, top5_accuracy:89.7 |
|
|
|
|
|
``` |
|
|
|
|
|
|
|
|
# [Model Description](#contents) |
|
|
# [Model Description](#contents) |
|
|
|
|
|
|
|
|
## [Performance](#contents) |
|
|
## [Performance](#contents) |
|
|
|