From a341e4f512d913d268d3209089efcc128dbdb29b Mon Sep 17 00:00:00 2001 From: Ji Hou Date: Tue, 11 Jun 2019 16:44:56 +0200 Subject: [PATCH] Update segmentation_mask.py number of instances should be in the first dimension. --- maskrcnn_benchmark/structures/segmentation_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/maskrcnn_benchmark/structures/segmentation_mask.py b/maskrcnn_benchmark/structures/segmentation_mask.py index 89b80aff2..af786787a 100644 --- a/maskrcnn_benchmark/structures/segmentation_mask.py +++ b/maskrcnn_benchmark/structures/segmentation_mask.py @@ -58,7 +58,7 @@ def __init__(self, masks, size): if len(masks) == 0: masks = torch.empty([0, size[1], size[0]]) # num_instances = 0! elif isinstance(masks[0], torch.Tensor): - masks = torch.stack(masks, dim=2).clone() + masks = torch.stack(masks, dim=0).clone() elif isinstance(masks[0], dict) and "counts" in masks[0]: # RLE interpretation rle_sizes = [tuple(inst["size"]) for inst in masks]