From 0de8baf186c8412e86d85852bdcdca443cbb5b14 Mon Sep 17 00:00:00 2001
From: CaoJian <caojian05>
Date: Mon, 4 Jan 2021 21:08:17 +0800
Subject: [PATCH] fix centernet data preprocess is slow.

---
 model_zoo/research/cv/centernet/src/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/model_zoo/research/cv/centernet/src/dataset.py b/model_zoo/research/cv/centernet/src/dataset.py
index 0b27397d47..608dd99e5c 100644
--- a/model_zoo/research/cv/centernet/src/dataset.py
+++ b/model_zoo/research/cv/centernet/src/dataset.py
@@ -32,6 +32,7 @@ from src.image import gaussian_radius, draw_umich_gaussian, draw_msra_gaussian,
 from src.visual import visual_image
 
 _current_dir = os.path.dirname(os.path.realpath(__file__))
+cv2.setNumThreads(0)
 
 
 class COCOHP(ds.Dataset):