mindspore-lab · lvyufeng · Feb 27, 2025 · Oct 23, 2024 · Oct 23, 2024 · Oct 23, 2024
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/applications/MaskFormer/Inference_with_MaskFormer_for_semantic_+_panoptic_segmentation.ipynb b/applications/MaskFormer/Inference_with_MaskFormer_for_semantic_+_panoptic_segmentation.ipynb
diff --git a/applications/MaskFormer/Minimal_example_of_MaskFormer_inference.ipynb b/applications/MaskFormer/Minimal_example_of_MaskFormer_inference.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]