diff --git a/official/cv/lenet/infer/convert/air2om.sh b/official/cv/lenet/infer/convert/air2om.sh index 642637d0fe153bfc108ea909412e3859e29e1fab..deefca83a34cafd37d2b80ebe1eafc5d51d5582f 100644 --- a/official/cv/lenet/infer/convert/air2om.sh +++ b/official/cv/lenet/infer/convert/air2om.sh @@ -16,7 +16,7 @@ model_path=$1 output_model_name=$2 -/usr/local/Ascend/atc/bin/atc \ +atc \ --model=$model_path \ --framework=1 \ --output=$output_model_name \ diff --git a/official/cv/lenet/infer/docker_start_infer.sh b/official/cv/lenet/infer/docker_start_infer.sh index be3a0b65ef06aa642a03197b9dec8b721317509a..1016a59b66512e4020e41fd7ab540dc51136f3cd 100644 --- a/official/cv/lenet/infer/docker_start_infer.sh +++ b/official/cv/lenet/infer/docker_start_infer.sh @@ -37,7 +37,7 @@ function param_check() { param_check -docker run -it \ +docker run -u root -it \ --device=/dev/davinci0 \ --device=/dev/davinci_manager \ --device=/dev/devmm_svm \ diff --git a/official/cv/lenet/infer/mxbase/LenetOpencv.cpp b/official/cv/lenet/infer/mxbase/LenetOpencv.cpp index ac875258f38fa2ea54bff8c401db8f3c61f2a16f..1e0272de6c1f4934b05dd08929a289556bbc9fcb 100644 --- a/official/cv/lenet/infer/mxbase/LenetOpencv.cpp +++ b/official/cv/lenet/infer/mxbase/LenetOpencv.cpp @@ -65,10 +65,10 @@ APP_ERROR LenetOpencv::DeInit() { return APP_ERR_OK; } -APP_ERROR LenetOpencv::ReadImage(const std::string &imgPath, cv::Mat &imageMat) { +void LenetOpencv::ReadImage(const std::string &imgPath, cv::Mat &imageMat) { imageMat = cv::imread(imgPath, cv::IMREAD_COLOR); } -APP_ERROR LenetOpencv::ResizeImage(const cv::Mat &srcImageMat, cv::Mat &dstImageMat) { +void LenetOpencv::ResizeImage(const cv::Mat &srcImageMat, cv::Mat &dstImageMat) { static constexpr uint32_t resizeHeight = 32; static constexpr uint32_t resizeWidth = 32; @@ -127,7 +127,7 @@ APP_ERROR LenetOpencv::Process(const std::string &imgPath) { std::vector<MxBase::TensorBase> inputs = {}; std::vector<MxBase::TensorBase> outputs = {}; TensorBase tensorBase; - ret = CVMatToTensorBase(imageMat, tensorBase); + APP_ERROR ret = CVMatToTensorBase(imageMat, tensorBase); if (ret != APP_ERR_OK) { LogError << "CVMatToTensorBase failed, ret=" << ret << "."; return ret; diff --git a/official/cv/lenet/infer/sdk/python_lenet/run.sh b/official/cv/lenet/infer/sdk/python_lenet/run.sh index a45b3e88a55b31498b6dd048f1f3b6cba8ccc664..7c1dad50ba6d32a8fe6eb0733263114d6d540b4c 100644 --- a/official/cv/lenet/infer/sdk/python_lenet/run.sh +++ b/official/cv/lenet/infer/sdk/python_lenet/run.sh @@ -33,6 +33,6 @@ warn() { echo >&2 -e "\033[1;31m[WARN ][MxStream] $1\033[1;37m" ; } info() { echo -e "\033[1;34m[INFO ][MxStream] $1\033[1;37m" ; } warn() { echo >&2 -e "\033[1;31m[WARN ][MxStream] $1\033[1;37m" ; } -python3.7 main.py $image_path $result_dir +python3 main.py $image_path $result_dir exit 0