fix: retinaface_resnet50 & psenet random get data timeout

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

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)