diff --git a/model_zoo/official/cv/psenet/src/dataset.py b/model_zoo/official/cv/psenet/src/dataset.py
index d6f8a8e961..d89f887b0a 100644
--- a/model_zoo/official/cv/psenet/src/dataset.py
+++ b/model_zoo/official/cv/psenet/src/dataset.py
@@ -172,6 +172,8 @@ def shrink(bboxes, rate, max_shr=20):
 
 class TrainDataset:
     def __init__(self):
+        cv2.setNumThreads(2)
+
         self.is_transform = True
         self.img_size = config.TRAIN_LONG_SIZE
         self.kernel_num = config.KERNEL_NUM
diff --git a/model_zoo/official/cv/retinaface_resnet50/src/dataset.py b/model_zoo/official/cv/retinaface_resnet50/src/dataset.py
index 716a8217ae..b786d97384 100644
--- a/model_zoo/official/cv/retinaface_resnet50/src/dataset.py
+++ b/model_zoo/official/cv/retinaface_resnet50/src/dataset.py
@@ -63,6 +63,7 @@ class WiderFace():
         return self.images_list[item], self.labels_list[item]
 
 def read_dataset(img_path, annotation):
+    cv2.setNumThreads(2)
 
     if isinstance(img_path, str):
         img = cv2.imread(img_path)