forked from mindspore-Ecosystem/mindspore
add ascend online convert model
This commit is contained in:
parent
f30f0b11d5
commit
ed41c0b614
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue