Browse Source

fix centernet data preprocess is slow.

tags/v1.2.0-rc1
CaoJian 4 years ago
parent
commit
76ba87f62f
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/research/cv/centernet/src/dataset.py

+ 1
- 0
model_zoo/research/cv/centernet/src/dataset.py View File

@@ -31,6 +31,7 @@ from src.image import gaussian_radius, draw_umich_gaussian, draw_msra_gaussian,
from src.visual import visual_image from src.visual import visual_image


_current_dir = os.path.dirname(os.path.realpath(__file__)) _current_dir = os.path.dirname(os.path.realpath(__file__))
cv2.setNumThreads(0)




class COCOHP(ds.Dataset): class COCOHP(ds.Dataset):


Loading…
Cancel
Save