!10940 fix centernet data preprocess is slow

From: @caojian05 Reviewed-by: @wuxuejian,@oacjiewen Signed-off-by: @wuxuejian
2021-01-05 10:14:54 +08:00 · 2021-01-05 10:14:54 +08:00 · f4b256d470
parent 90bd96c9f6 76ba87f62f
commit f4b256d470
1 changed files with 1 additions and 0 deletions
--- 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):