!18520 set opencv thread number for dpn

Merge pull request !18520 from zhouneng/fix_issue_I3J4XD

model_zoo/official/cv/dpn/eval.py

@@ -62,8 +62,7 @@ def dpn_evaluate():
         loss = CrossEntropy(smooth_factor=config.label_smooth_factor, num_classes=config.num_classes)
 
         # create model
-    model = Model(net, amp_level="O2", keep_batchnorm_fp32=False, loss_fn=loss,
-                  metrics={'top_1_accuracy', 'top_5_accuracy'})
+    model = Model(net, keep_batchnorm_fp32=False, loss_fn=loss, metrics={'top_1_accuracy', 'top_5_accuracy'})
     # evaluate
     output = model.eval(eval_dataset)
     print(f'Evaluation result: {output}.')

model_zoo/official/cv/dpn/src/imagenet_dataset.py

@@ -15,6 +15,8 @@
 """
 dataset processing.
 """
+import cv2
+
 from PIL import ImageFile
 from mindspore.common import dtype as mstype
 import mindspore.dataset as de
@@ -74,6 +76,7 @@ def classification_dataset(data_dir, image_size, per_batch_size, max_epoch, rank
         >>>                               rank=0, group_size=4,
         >>>                               input_mode="txt", root=images_dir)
     """
+    cv2.setNumThreads(0)
     if mode == 'eval':
         drop_remainder = False
     else: