From d4383c4f33aa0101cf50556eaa7793da441f37d3 Mon Sep 17 00:00:00 2001 From: yanghaitao1 Date: Fri, 9 Oct 2020 10:49:35 +0800 Subject: [PATCH] set opencv threads 0 --- mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc index 8094734915..93190b1b0c 100644 --- a/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc +++ b/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc @@ -97,6 +97,8 @@ Status Resize(const std::shared_ptr &input, std::shared_ptr *out if (input_cv->Rank() != 3 && input_cv->Rank() != 2) { RETURN_STATUS_UNEXPECTED("Input Tensor is not in shape of or "); } + // OpenCv lanuch too many threads. + cv::setNumThreads(0); cv::Mat in_image = input_cv->mat(); // resize image too large or too small if (output_height == 0 || output_height > in_image.rows * 1000 || output_width == 0 ||