From ba763567028b37e6e01fb7a519c8bd8b7e2b40ea Mon Sep 17 00:00:00 2001 From: jonyguo Date: Wed, 6 Jan 2021 09:17:39 +0800 Subject: [PATCH] fix: retinaface_resnet50 & psenet random get data timeout --- model_zoo/official/cv/psenet/src/dataset.py | 2 ++ model_zoo/official/cv/retinaface_resnet50/src/dataset.py | 1 + 2 files changed, 3 insertions(+) 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)