From db1e522e2da7d785cc35bf732ff3e7e7ab6cc35b Mon Sep 17 00:00:00 2001 From: wukesong Date: Fri, 11 Sep 2020 16:01:30 +0800 Subject: [PATCH] limit cv thread num. --- model_zoo/official/cv/deeplabv3/src/data/data_generator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/official/cv/deeplabv3/src/data/data_generator.py b/model_zoo/official/cv/deeplabv3/src/data/data_generator.py index 8e58d46ded6..4cbcafa4254 100644 --- 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: