diff --git a/local_validate_model.sh b/local_validate_model.sh index 1c2eeb1aa126b827c3266e1de4131276832712c5..8a9d50ba0e2d995e3fe26de951aceb7a8f58d606 100755 --- a/local_validate_model.sh +++ b/local_validate_model.sh @@ -89,20 +89,20 @@ python tools/validate.py --generate_data true \ --input_shape="${INPUT_SHAPE}" || exit -1 echo "Step 2: Convert tf model to mace model and optimize memory" -#bazel build //lib/python/tools:tf_converter || exit -1 -#rm -rf ${MODEL_CODEGEN_DIR} -#mkdir -p ${MODEL_CODEGEN_DIR} -#bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ -# --output=${MODEL_CODEGEN_DIR}/model.cc \ -# --input_node=${TF_INPUT_NODE} \ -# --output_node=${TF_OUTPUT_NODE} \ -# --data_type=${DATA_TYPE} \ -# --runtime=${RUNTIME} \ -# --output_type=source \ -# --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \ -# --model_tag=${MODEL_TAG} \ -# --input_shape="${INPUT_SHAPE}" \ -# --obfuscate=False || exit -1 +bazel build //lib/python/tools:tf_converter || exit -1 +rm -rf ${MODEL_CODEGEN_DIR} +mkdir -p ${MODEL_CODEGEN_DIR} +bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ + --output=${MODEL_CODEGEN_DIR}/model.cc \ + --input_node=${TF_INPUT_NODE} \ + --output_node=${TF_OUTPUT_NODE} \ + --data_type=${DATA_TYPE} \ + --runtime=${RUNTIME} \ + --output_type=source \ + --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \ + --model_tag=${MODEL_TAG} \ + --input_shape="${INPUT_SHAPE}" \ + --obfuscate=False || exit -1 echo "Step 3: Download mace static library" download_and_link_lib