air to mindir

official/cv/alexnet/config_imagenet.yaml

@@ -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

official/cv/alexnet/default_config.yaml

@@ -38,7 +38,7 @@ lr: 0.01
 # Model Description
 model_name: alexnet
 file_name: 'alexnet'
-file_format: 'AIR'
+file_format: 'MINDIR'
 
 
 ---

official/cv/centerface/default_config.yaml

@@ -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: ''

official/cv/deeplabv3/default_config.yaml

@@ -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"
 
 ---

official/cv/faster_rcnn/default_config.yaml

@@ -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")

official/cv/faster_rcnn/default_config_101.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")

official/cv/faster_rcnn/default_config_152.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")

official/cv/faster_rcnn/default_config_InceptionResnetV2.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")

official/cv/maskrcnn/default_config.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'
 

official/cv/vgg16/cifar10_config.yaml

@@ -67,7 +67,7 @@ result_path: "./preprocess_Result/"
 # export options
 ckpt_file: ""
 file_name: "vgg16"
-file_format: "AIR"
+file_format: "MINDIR"
 
 
 ---

official/cv/vgg16/imagenet2012_config.yaml

@@ -67,7 +67,7 @@ result_path: "./preprocess_Result/"
 # export options
 ckpt_file: ""
 file_name: "vgg16"
-file_format: "AIR"
+file_format: "MINDIR"
 
 ---
 

official/cv/yolov3_darknet53/default_config.yaml

@@ -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

official/cv/yolov3_resnet18/default_config.yaml

@@ -38,7 +38,7 @@ device_id: 0
 export_batch_size: 1
 ckpt_file: ""
 file_name: "yolov3_resnet18"
-file_format: "AIR"
+file_format: "MINDIR"
 
 ---
 

official/cv/yolov4/default_config.yaml

@@ -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

official/nlp/lstm/README.md

@@ -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)
 

official/nlp/lstm/README_CN.md

@@ -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
 
 ### 结果