fix centernet data preprocess is slow.

4 years ago · 76ba87f62f
parent 3d7cb8db4c
commit 76ba87f62f
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):