Unverified Commit 6855f237 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub

Fix demo after PostProcess mask (#187)

parent ca9531b9
...@@ -211,7 +211,8 @@ class COCODemo(object): ...@@ -211,7 +211,8 @@ class COCODemo(object):
# if we have masks, paste the masks in the right position # if we have masks, paste the masks in the right position
# in the image, as defined by the bounding boxes # in the image, as defined by the bounding boxes
masks = prediction.get_field("mask") 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) prediction.add_field("mask", masks)
return prediction return prediction
......
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