!6053 limit_cv_threads_num_to_deeplabv3

Merge pull request !6053 from wukesong/limit_cv_for_deeplabv3
2020-09-12 15:22:35 +08:00 · 2020-09-12 15:22:35 +08:00 · daf6739b22
parent ad37b6845f db1e522e2d
commit daf6739b22
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/official/cv/deeplabv3/src/data/data_generator.py
+++ b/model_zoo/official/cv/deeplabv3/src/data/data_generator.py
@ -16,6 +16,7 @@
 import cv2
 import numpy as np
 import mindspore.dataset as de
+cv2.setNumThreads(0)


 class SegDataset: