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