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):