limit cv thread num.

5 years ago · db1e522e2d
parent acd896cdea
commit db1e522e2d
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: