Add RLE support (#657)

https://gist.github.com/botcs/a59f3f59e22e5df93e3e5e4f86718af3
facebookresearch · Apr 13, 2019 · c5c4d52 · c5c4d52
1 parent 42fa3a1
commit c5c4d52
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/maskrcnn_benchmark/structures/segmentation_mask.py b/maskrcnn_benchmark/structures/segmentation_mask.py
@@ -54,10 +54,13 @@ def __init__(self, masks, size):
         elif isinstance(masks, (list, tuple)):
             if isinstance(masks[0], torch.Tensor):
                 masks = torch.stack(masks, dim=2).clone()
-            elif isinstance(masks[0], dict) and "count" in masks[0]:
+            elif isinstance(masks[0], dict) and "counts" in masks[0]:
                 # RLE interpretation
-
-                masks = mask_utils
+                assert all(
+                    [(size[0], size[1]) == tuple(inst["size"]) for inst in masks]
+                )
+                masks = mask_utils.decode(masks)
+                masks = torch.tensor(masks).permute(2, 0, 1)
             else:
                 RuntimeError(
                     "Type of `masks[0]` could not be interpreted: %s" % type(masks)