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

4 years ago · 0a1d524404
parent 995e973fec
commit 0a1d524404
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/official/cv/ssd/src/dataset.py
+++ b/model_zoo/official/cv/ssd/src/dataset.py
@ -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