Browse Source

fix: when ssd net, high cpu usage & get_next err

tags/v1.1.0
jonyguo 5 years ago
parent
commit
0a1d524404
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/official/cv/ssd/src/dataset.py

+ 1
- 0
model_zoo/official/cv/ssd/src/dataset.py View File

@@ -105,6 +105,7 @@ def random_sample_crop(image, boxes):

def preprocess_fn(img_id, image, box, is_training):
"""Preprocess function for dataset."""
cv2.setNumThreads(2)

def _infer_data(image, input_shape):
img_h, img_w, _ = image.shape


Loading…
Cancel
Save