From fd022b931ab9d8bae6a9933121dca6a80d16c9d9 Mon Sep 17 00:00:00 2001
From: zhouneng2 <zhouneng2@huawei.com>
Date: Tue, 15 Feb 2022 17:55:43 +0800
Subject: [PATCH] [FasterRCNN]fix import error

---
 official/cv/faster_rcnn/eval.py               | 1 +
 official/cv/faster_rcnn/src/network_define.py | 6 +++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/official/cv/faster_rcnn/eval.py b/official/cv/faster_rcnn/eval.py
index 0e9294dec..86da7c5a7 100644
--- a/official/cv/faster_rcnn/eval.py
+++ b/official/cv/faster_rcnn/eval.py
@@ -20,6 +20,7 @@ from collections import defaultdict
 
 import numpy as np
 from pycocotools.coco import COCO
+import mindspore as ms
 from mindspore.common import set_seed, Parameter
 
 from src.dataset import data_to_mindrecord_byte_image, create_fasterrcnn_dataset, parse_json_annos_from_txt
diff --git a/official/cv/faster_rcnn/src/network_define.py b/official/cv/faster_rcnn/src/network_define.py
index c7edbf325..cf62aa6ba 100644
--- a/official/cv/faster_rcnn/src/network_define.py
+++ b/official/cv/faster_rcnn/src/network_define.py
@@ -15,10 +15,10 @@
 """FasterRcnn training network wrapper."""
 
 import time
-import numpy as np
+import mindspore.common.dtype as mstype
 import mindspore.ops as ops
 import mindspore.nn as nn
-from mindspore import ParameterTuple
+from mindspore import ParameterTuple, Tensor
 from mindspore.train.callback import Callback
 from mindspore.nn.wrap.grad_reducer import DistributedGradReducer
 
@@ -134,7 +134,7 @@ class TrainOneStepCell(nn.Cell):
         self.optimizer = optimizer
         self.grad = ops.GradOperation(get_by_list=True,
                                       sens_param=True)
-        self.sens = ms.numpy.ones((1,) * sens).astype(np.float32)
+        self.sens = Tensor([sens,], mstype.float32)
         self.reduce_flag = reduce_flag
         if reduce_flag:
             self.grad_reducer = DistributedGradReducer(optimizer.parameters, mean, degree)
-- 
GitLab