Browse Source

!6053 limit_cv_threads_num_to_deeplabv3

Merge pull request !6053 from wukesong/limit_cv_for_deeplabv3
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
daf6739b22
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/official/cv/deeplabv3/src/data/data_generator.py

+ 1
- 0
model_zoo/official/cv/deeplabv3/src/data/data_generator.py View File

@@ -16,6 +16,7 @@
import cv2 import cv2
import numpy as np import numpy as np
import mindspore.dataset as de import mindspore.dataset as de
cv2.setNumThreads(0)
class SegDataset: class SegDataset:


Loading…
Cancel
Save