Browse Source

!7287 [MD] minddata fix opencv float exception issue

Merge pull request !7287 from xiefangqi/md_fix_opencv_issue
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
bcd72c25ee
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc

+ 1
- 1
mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc View File

@@ -98,7 +98,7 @@ Status Resize(const std::shared_ptr<Tensor> &input, std::shared_ptr<Tensor> *out
RETURN_STATUS_UNEXPECTED("Input Tensor is not in shape of <H,W,C> or <H,W>");
}
// OpenCv lanuch too many threads.
cv::setNumThreads(0);
cv::setNumThreads(8);
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 ||


Loading…
Cancel
Save