Browse Source

!9785 fix ssd net get next err

From: @jonyguo
Reviewed-by: @liucunwei,@pandoublefeng
Signed-off-by: @liucunwei
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
8186e6ef9e
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