!13782 update example and run script

From: @zhujingxuan
Reviewed-by: @wangchengyuan,@wangchengyuan,@zhanghaibo5
Signed-off-by: @wangchengyuan,@wangchengyuan

mindspore/lite/micro/example/mnist_x86/mnist.sh

@@ -0,0 +1,94 @@
+#!/bin/bash
+# Copyright 2021 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+set -e
+
+GEN=OFF
+while getopts 'g' OPT
+do
+    case $OPT in
+        g)
+            GEN=ON;;
+        ?)
+            echo "Usage: add -g or left it empty"
+    esac
+done
+
+BASEPATH="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
+MINDSPORE_ROOT_DIR=${${BASEPATH}%%/mindspore/lite/micro/example/mnist}
+
+echo "current dir is: ${BASEPATH}"
+
+INPUT_BIN=${BASEPATH}/mnist_input.bin
+BENCHMARK_PATH=${BASEPATH}
+
+get_version() {
+    local VERSION_HEADER=${MINDSPORE_ROOT_DIR}/mindspore/lite/include/version.h
+    local VERSION_MAJOR=$(grep "const int ms_version_major =" ${VERSION_HEADER} | tr -dc "[0-9]")
+    local VERSION_MINOR=$(grep "const int ms_version_minor =" ${VERSION_HEADER} | tr -dc "[0-9]")
+    local VERSION_REVISION=$(grep "const int ms_version_revision =" ${VERSION_HEADER} | tr -dc "[0-9]")
+    VERSION_STR=${VERSION_MAJOR}.${VERSION_MINOR}.${VERSION_REVISION}
+}
+
+download_inference() {
+    local MINDSPORE_FILE_NAME="mindspore-lite-${VERSION_STR}-inference-linux-x64"
+    local MINDSPORE_FILE="${MINDSPORE_FILE_NAME}.tar.gz"
+    local MINDSPORE_LITE_DOWNLOAD_URL="https://ms-release.obs.cn-north-4.myhuaweicloud.com/${VERSION_STR}/MindSpore/lite/release/linux/${MINDSPORE_FILE}"
+
+    if [ ! -e ${BASEPATH}/build/${MINDSPORE_FILE} ]; then
+      wget -c -O ${BASEPATH}/build/${MINDSPORE_FILE} --no-check-certificate ${MINDSPORE_LITE_DOWNLOAD_URL}
+    fi
+
+    tar xzvf ${BASEPATH}/build/${MINDSPORE_FILE} -C ${BASEPATH}/build/ || exit 1
+    rm ${BASEPATH}/build/${MINDSPORE_FILE} || exit 1
+    PKG_PATH=${BASEPATH}/build/${MINDSPORE_FILE_NAME}
+}
+
+download_mnist() {
+    MNIST_NAME=mnist
+    MNIST_FILE=${MNIST_NAME}.ms
+    local MNIST_DOWNLOAD_URL=https://download.mindspore.cn/model_zoo/official/lite/mnist_lite/${MNIST_FILE}
+
+    if [ ! -e ${BASEPATH}/build/${MNIST_FILE} ]; then
+      wget -c -O ${BASEPATH}/build/${MNIST_FILE} --no-check-certificate ${MNIST_DOWNLOAD_URL}
+    fi
+}
+
+gen_mnist() {
+    local CODEGEN_PATH=${BASEPATH}/build/${MINDSPORE_FILE_NAME}/tools/codegen
+    ${CODEGEN_PATH}/codegen --codePath=${BASEPATH}/build --modelPath=${BASEPATH}/build/${MNIST_FILE}
+}
+
+mkdir -p build
+
+get_version
+download_inference
+
+if [ "${GEN}" == "ON" ]; then
+    echo "downloading mnist.ms!"
+    download_mnist
+    echo "generating mnist"
+    gen_mnist
+    BENCHMARK_PATH=${BASEPATH}/build/${MNIST_NAME}
+fi
+
+# 1. build benchmark
+mkdir -p ${BASEPATH}/build/benchmark && cd ${BASEPATH}/build/benchmark || exit 1
+cmake -DPKG_PATH=${PKG_PATH} ${BENCHMARK_PATH}
+make
+
+# 2. run benchmark
+echo "net file: ${BENCHMARK_PATH}/src/mnist.bin"
+./benchmark ${INPUT_BIN} ${BENCHMARK_PATH}/src/net.bin

mindspore/lite/micro/example/mnist/mnist.sh → mindspore/lite/micro/example/mnist_x86/mnist_codegen.sh

@@ -34,20 +34,32 @@ MINDSPORE_FILE_NAME="mindspore-lite-${VERSION_STR}-inference-linux-x64"
 MINDSPORE_FILE="${MINDSPORE_FILE_NAME}.tar.gz"
 MINDSPORE_LITE_DOWNLOAD_URL="https://ms-release.obs.cn-north-4.myhuaweicloud.com/${VERSION_STR}/MindSpore/lite/release/linux/${MINDSPORE_FILE}"
 
+MNIST_NAME=mnist
+MNIST_FILE=${MNIST_NAME}.ms
+MNIST_DOWNLOAD_URL=https://download.mindspore.cn/model_zoo/official/lite/mnist_lite/${MNIST_FILE}
+
 mkdir -p build
 
 if [ ! -e ${BASEPATH}/build/${MINDSPORE_FILE} ]; then
   wget -c -O ${BASEPATH}/build/${MINDSPORE_FILE} --no-check-certificate ${MINDSPORE_LITE_DOWNLOAD_URL}
 fi
 
+if [ ! -e ${BASEPATH}/build/${MNIST_FILE} ]; then
+  wget -c -O ${BASEPATH}/build/${MNIST_FILE} --no-check-certificate ${MNIST_DOWNLOAD_URL}
+fi
+
 tar xzvf ${BASEPATH}/build/${MINDSPORE_FILE} -C ${BASEPATH}/build/ || exit 1
 rm ${BASEPATH}/build/${MINDSPORE_FILE} || exit 1
 PKG_PATH=${BASEPATH}/build/${MINDSPORE_FILE_NAME}
-# build benchmark
+
+# 1. codegen
+${BASEPATH}/build/${MINDSPORE_FILE_NAME}/tools/codegen/codegen --codePath=${BASEPATH}/build --modelPath=${BASEPATH}/build/${MNIST_FILE}
+
+# 2. build benchmark
 mkdir -p ${BASEPATH}/build/benchmark && cd ${BASEPATH}/build/benchmark || exit 1
-cmake -DPKG_PATH=${PKG_PATH} ${BASEPATH}
+cmake -DPKG_PATH=${PKG_PATH} ${BASEPATH}/build/${MNIST_NAME}
 make
 
-echo "net file: ${BASEPATH}/src/mnist.bin"
 # 3. run benchmark
+echo "net file: ${BASEPATH}/src/mnist.bin"
 ./benchmark ${INPUT_BIN} ${BASEPATH}/src/net.bin