[Fix] Fix ascend faster-rcnn (#1842)

* Fix ascend faster-rcnn

* remove static config

Author: grimoire

configs/mmdet/detection/detection_ascend_static-800x1344.py

@@ -394,7 +394,7 @@ def forward(self, inputs: Dict[str,
 
             for binding in self._model_desc.outputs:
                 self._copy_buffer_to_tensor(
-                    self._output.buffers[binding.index], tensor)
+                    self._output.buffers[binding.index], outputs[binding.name])
 
             return outputs
 

mmdeploy/backend/ascend/wrapper.py

@@ -249,6 +249,9 @@ def __gather_topk__trt(*inputs: Sequence[torch.Tensor],
     return outputs
 
 
+@FUNCTION_REWRITER.register_rewriter(
+    'mmdeploy.codebase.mmdet.deploy.utils.__gather_topk',
+    backend=Backend.ASCEND.value)
 @FUNCTION_REWRITER.register_rewriter(
     'mmdeploy.codebase.mmdet.deploy.utils.__gather_topk',
     backend=Backend.COREML.value)

mmdeploy/codebase/mmdet/deploy/utils.py

@@ -38,7 +38,7 @@ def standard_roi_head__predict_bbox(self,
                 (num_instances, ).
     """
     rois = rpn_results_list[0]
-    rois_dims = rois.shape[-1]
+    rois_dims = int(rois.shape[-1])
     batch_index = torch.arange(
         rois.shape[0], device=rois.device).float().view(-1, 1, 1).expand(
             rois.size(0), rois.size(1), 1)

mmdeploy/codebase/mmdet/models/roi_heads/standard_roi_head.py

@@ -713,4 +713,4 @@ def multiclass_nms__ascend(boxes: Tensor,
         boxes, scores, score_threshold, iou_threshold, keep_top_k, keep_top_k)
 
     dets = torch.cat([nmsed_boxes, nmsed_scores.unsqueeze(2)], dim=-1)
-    return dets, nmsed_classes
+    return dets, nmsed_classes.int()