【开源实习】Mask2Former模型应用开发 (#1770)

mindspore-lab · Feb 27, 2025 · 1bd7e81 · 1bd7e81
1 parent 8c26b18
commit 1bd7e81
Show file tree

Hide file tree

Showing 3 changed files with 1,109 additions and 3 deletions.
diff --git a/...s/AST/Audio Spectrogram Transformer.ipynb → ...s/AST/Audio Spectrogram Transformer.ipynb b/...s/AST/Audio Spectrogram Transformer.ipynb → ...s/AST/Audio Spectrogram Transformer.ipynb
diff --git a/applications/Mask2Former/Inference_with_Mask2Former.ipynb b/applications/Mask2Former/Inference_with_Mask2Former.ipynb
diff --git a/mindnlp/transformers/models/mask2former/image_processing_mask2former.py b/mindnlp/transformers/models/mask2former/image_processing_mask2former.py
@@ -59,7 +59,6 @@
 
 if is_mindspore_available():
     import mindspore
-    from mindspore import nn
 
 
 # Copied from transformers.models.detr.image_processing_detr.max_across_indices
@@ -215,7 +214,7 @@ def compute_segments(
     segments: List[Dict] = []
 
     if target_size is not None:
-        mask_probs = nn.functional.interpolate(
+        mask_probs = ops.interpolate(
             mask_probs.unsqueeze(0), size=target_size, mode="bilinear", align_corners=False
         )[0]
 
@@ -1103,7 +1102,7 @@ def post_process_instance_segmentation(
             mask_cls = class_queries_logits[i]
 
             scores = ops.softmax(mask_cls, axis=-1)[:, :-1]
-            labels = ops.arange(num_classes).unsqueeze(0).repeat(num_queries, 1).flatten()
+            labels = ops.arange(num_classes).unsqueeze(0).tile((num_queries, 1)).flatten()
 
             scores_per_image, topk_indices = scores.flatten().topk(num_queries, sorted=False)
             labels_per_image = labels[topk_indices]