Fix demo after PostProcess mask (#187)

facebookresearch · Nov 20, 2018 · 6855f23 · 6855f23
1 parent ca9531b
commit 6855f23
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/demo/predictor.py b/demo/predictor.py
@@ -211,7 +211,8 @@ def compute_prediction(self, original_image):
             # if we have masks, paste the masks in the right position
             # in the image, as defined by the bounding boxes
             masks = prediction.get_field("mask")
-            masks = self.masker(masks, prediction)
+            # always single image is passed at a time
+            masks = self.masker([masks], [prediction])[0]
             prediction.add_field("mask", masks)
         return prediction