Remove -1 from crop mask

Summary: Removing 1 from the crop mask does not seem sensible.

Reviewed By: bottler, shapovalov

Differential Revision: D37843680

fbshipit-source-id: 70cec80f9ea26deac63312da62b9c8af27d2a010
This commit is contained in:
David Novotny 2022-07-14 03:30:51 -07:00 committed by Facebook GitHub Bot
parent 36ba079bef
commit 8ba9a694ee

View File

@ -779,7 +779,7 @@ class JsonIndexDataset(DatasetBase, ReplaceableBase):
# pyre-fixme[6]: For 2nd param expected `int` but got `Optional[int]`.
# pyre-fixme[6]: For 3rd param expected `int` but got `Optional[int]`.
mask = torch.zeros(1, self.image_height, self.image_width)
mask[:, 0 : imre.shape[1] - 1, 0 : imre.shape[2] - 1] = 1.0
mask[:, 0 : imre.shape[1], 0 : imre.shape[2]] = 1.0
return imre_, minscale, mask
def _local_path(self, path: str) -> str: