From 522e5f06441b7faa9b5c474ef6e016398710eb14 Mon Sep 17 00:00:00 2001 From: David Novotny Date: Sun, 10 Jul 2022 16:01:33 -0700 Subject: [PATCH] Bugfix - wrong mask bounds passed to box clamping Summary: Fixes a bug Reviewed By: bottler Differential Revision: D37743350 fbshipit-source-id: d68e680d6027ae2b9814b2241fb72d3b74df77c1 --- pytorch3d/implicitron/dataset/json_index_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch3d/implicitron/dataset/json_index_dataset.py b/pytorch3d/implicitron/dataset/json_index_dataset.py index b1640a13..6b3d7102 100644 --- a/pytorch3d/implicitron/dataset/json_index_dataset.py +++ b/pytorch3d/implicitron/dataset/json_index_dataset.py @@ -427,7 +427,6 @@ class JsonIndexDataset(DatasetBase, ReplaceableBase): image_path=entry.image.path, box_crop_context=self.box_crop_context, ) - mask = _crop_around_box(mask, clamp_bbox_xyxy, full_path) crop_box_xyxy = _clamp_box_to_image_bounds_and_round( clamp_bbox_xyxy, @@ -435,6 +434,8 @@ class JsonIndexDataset(DatasetBase, ReplaceableBase): ) crop_box_xywh = _bbox_xyxy_to_xywh(crop_box_xyxy) + mask = _crop_around_box(mask, clamp_bbox_xyxy, full_path) + fg_probability, _, _ = self._resize_image(mask, mode="nearest") return fg_probability, full_path, bbox_xywh, clamp_bbox_xyxy, crop_box_xywh