From 76ba87f62fec4a28a792d3a267d0c45835fe953f 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 b3f5f8eb61..b67dc5dc8d 100644
--- a/model_zoo/research/cv/centernet/src/dataset.py
+++ b/model_zoo/research/cv/centernet/src/dataset.py
@@ -31,6 +31,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):