Browse Source

limit cv thread num.

tags/v1.0.0
wukesong 5 years ago
parent
commit
db1e522e2d
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