Skip to content
Snippets Groups Projects
Commit 1bc514df authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!1316 change settings air to mindir

Merge pull request !1316 from 张晓晓/master
parents 05808526 08b28044
No related branches found
No related tags found
No related merge requests found
Showing
with 25 additions and 22 deletions
......@@ -34,7 +34,7 @@ is_dynamic_loss_scale: 0
# Model Description
model_name: alexnet
file_name: 'alexnet'
file_format: 'AIR'
file_format: 'MINDIR'
dataset_name: 'imagenet'
sink_size: -1
......
......@@ -38,7 +38,7 @@ lr: 0.01
# Model Description
model_name: alexnet
file_name: 'alexnet'
file_format: 'AIR'
file_format: 'MINDIR'
---
......
......@@ -126,7 +126,7 @@ device_id: 0
batch_size: 1
ckpt_file: ''
file_name: "centerface"
file_format: 'AIR'
file_format: 'MINDIR'
# centerface preprocess"
dataset_path: ''
......
......@@ -63,7 +63,7 @@ export_batch_size: 1
input_size: 513
ckpt_file: ""
file_name: "deeplabv3"
file_format: "AIR"
file_format: "MINDIR"
export_model: "deeplab_v3_s8"
---
......
......@@ -166,7 +166,7 @@ checkpoint_path: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# export.py fasterrcnn_export
file_name: "faster_rcnn"
file_format: "AIR"
file_format: "MINDIR"
ckpt_file: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# postprocess ("./src/config_50.yaml")
......
......@@ -167,7 +167,7 @@ checkpoint_path: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# export.py fasterrcnn_export
file_name: "faster_rcnn"
file_format: "AIR"
file_format: "MINDIR"
ckpt_file: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# postprocess ("./src/config_50.yaml")
......
......@@ -167,7 +167,7 @@ checkpoint_path: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# export.py fasterrcnn_export
file_name: "faster_rcnn"
file_format: "AIR"
file_format: "MINDIR"
ckpt_file: "/cache/train/fasterrcnn/faster_rcnn-12_7393.ckpt"
# postprocess ("./src/config_50.yaml")
......
......@@ -165,7 +165,7 @@ checkpoint_path: "/cache/checkpoint_path/faster_rcnn-20_7393.ckpt"
# export.py fasterrcnn_export
file_name: "faster_rcnn"
file_format: "AIR"
file_format: "MINDIR"
ckpt_file: ""
# postprocess ("./src/default_config_InceptionResnetV2.yaml")
......
......@@ -179,7 +179,7 @@ rank_id: 0
# maskrcnn export
file_name: "maskrcnn"
file_format: "AIR"
file_format: "MINDIR"
ckpt_file: '/cache/data/cocodataset/ckpt_maskrcnn/mask_rcnn-12_7393.ckpt'
ckpt_file_local: './maskrcnn/scripts/train_parallel0/ckpt_0/mask_rcnn-12_7393.ckpt'
......
......@@ -67,7 +67,7 @@ result_path: "./preprocess_Result/"
# export options
ckpt_file: ""
file_name: "vgg16"
file_format: "AIR"
file_format: "MINDIR"
---
......
......@@ -67,7 +67,7 @@ result_path: "./preprocess_Result/"
# export options
ckpt_file: ""
file_name: "vgg16"
file_format: "AIR"
file_format: "MINDIR"
---
......
......@@ -74,7 +74,7 @@ device_id: 0
batch_size: 1
ckpt_file: ""
file_name: "yolov3_darknet53"
file_format: "AIR" # ["AIR", "ONNX", "MINDIR"]
file_format: "MINDIR" # ["AIR", "ONNX", "MINDIR"]
keep_detect: True
# PostProcess option
......
......@@ -38,7 +38,7 @@ device_id: 0
export_batch_size: 1
ckpt_file: ""
file_name: "yolov3_resnet18"
file_format: "AIR"
file_format: "MINDIR"
---
......
......@@ -63,7 +63,7 @@ batch_size: 1
testing_shape: 608
ckpt_file: ""
file_name: "yolov4"
file_format: "AIR"
file_format: "MINDIR"
keep_detect: True
# Other default config
......
......@@ -399,11 +399,12 @@ Ascend:
## [Export MindIR](#contents)
```shell
python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT]
python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] --config_path [YAML_CONFIG_PATH]
```
The ckpt_file parameter is required,
`EXPORT_FORMAT` should be in ["AIR", "MINDIR"]
- `ckpt_file` parameter is required.
- `FILE_FORMAT` should be in ["AIR", "MINDIR"].
- `YAML_CONFIG_PATH` default is `default_config.yaml`.
## [Inference Process](#contents)
......
......@@ -400,10 +400,12 @@ Ascend:
## 导出mindir模型
```shell
python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT]
python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] --config_path [YAML_CONFIG_PATH]
```
参数`ckpt_file` 是必需的,`EXPORT_FORMAT` 必须在 ["AIR", "MINDIR"]中进行选择。
- `ckpt_file` 是必需的。
- `FILE_FORMAT` 必须在 ["AIR", "MINDIR"]中进行选择。
- `YAML_CONFIG_PATH` 默认是 `default_config.yaml`
## 推理过程
......@@ -416,9 +418,9 @@ python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [
bash run_infer_310.sh [MINDIR_PATH] [DATASET_PATH] [NEED_PREPROCESS] [DEVICE_TARGET] [DEVICE_ID]
```
`DEVICE_TARGET` 可选值范围为:['GPU', 'CPU', 'Ascend']
`NEED_PREPROCESS` 表示数据是否需要预处理,可选值范围为:'y' 或者 'n'
`DEVICE_ID` 可选, 默认值为0.
- `DEVICE_TARGET` 可选值范围为:['GPU', 'CPU', 'Ascend']
- `NEED_PREPROCESS` 表示数据是否需要预处理,可选值范围为:'y' 或者 'n'
- `DEVICE_ID` 可选, 默认值为0
### 结果
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment