Unverified Commit 80f29a25 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

MaskFormer post_process_instance_segmentation bug fix convert out side of loop (#25497)

Bug fix - convert out side of loop
parent ee7d6694
...@@ -1026,9 +1026,10 @@ class Mask2FormerImageProcessor(BaseImageProcessor): ...@@ -1026,9 +1026,10 @@ class Mask2FormerImageProcessor(BaseImageProcessor):
) )
current_segment_id += 1 current_segment_id += 1
instance_maps.append(pred_masks[j]) instance_maps.append(pred_masks[j])
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation: # Return segmentation map in run-length encoding (RLE) format
segmentation = convert_segmentation_to_rle(segmentation) if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return a concatenated tensor of binary instance maps # Return a concatenated tensor of binary instance maps
if return_binary_maps and len(instance_maps) != 0: if return_binary_maps and len(instance_maps) != 0:
......
...@@ -1081,9 +1081,10 @@ class MaskFormerImageProcessor(BaseImageProcessor): ...@@ -1081,9 +1081,10 @@ class MaskFormerImageProcessor(BaseImageProcessor):
) )
current_segment_id += 1 current_segment_id += 1
instance_maps.append(pred_masks[j]) instance_maps.append(pred_masks[j])
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation: # Return segmentation map in run-length encoding (RLE) format
segmentation = convert_segmentation_to_rle(segmentation) if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return a concatenated tensor of binary instance maps # Return a concatenated tensor of binary instance maps
if return_binary_maps and len(instance_maps) != 0: if return_binary_maps and len(instance_maps) != 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment