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):
)
current_segment_id += 1
instance_maps.append(pred_masks[j])
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return a concatenated tensor of binary instance maps
if return_binary_maps and len(instance_maps) != 0:
......
......@@ -1081,9 +1081,10 @@ class MaskFormerImageProcessor(BaseImageProcessor):
)
current_segment_id += 1
instance_maps.append(pred_masks[j])
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return segmentation map in run-length encoding (RLE) format
if return_coco_annotation:
segmentation = convert_segmentation_to_rle(segmentation)
# Return a concatenated tensor of binary instance maps
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