add ascend online convert model

This commit is contained in:
wangwenzhe 2022-11-29 16:56:14 +08:00
parent f30f0b11d5
commit ed41c0b614
2 changed files with 8 additions and 0 deletions

View File

@ -11,6 +11,11 @@ EI_yolox.onnx;1:input;1,3,640,640;;offline_resize 5
EI_screenshot_text_location_220613_100.onnx;1:input_images;1,3,1024,1024;;offline_resize 5
EI_resnet18_batch_size1.onnx;1:input_node;1,6,224,224;;offline_resize 5
#online converter
EI_yolox.onnx;1:input;1,3,640,640;;online_convert 5
EI_screenshot_text_location_220613_100.onnx;1:input_images;1,3,1024,1024;;online_convert 5
EI_resnet18_batch_size1.onnx;1:input_node;1,6,224,224;;online_convert 5
#mindir
mindspore_uniir_mobilenetv2.mindir 0.5

View File

@ -90,6 +90,9 @@ function Convert() {
fi
export_mindir="MINDIR"
target_device="Ascend310"
if [[ ${extra_info} =~ "online_convert" ]]; then
target_device=""
fi
if [ ${model_fmk} != "TF" ]; then
input_format="NHWC"
fi