update interpolate

AyushExel · Sep 12, 2022 · 5a9d410 · 5a9d410
1 parent 9d6fed1
commit 5a9d410
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/segment/loss.py b/utils/segment/loss.py
@@ -83,7 +83,7 @@ def __call__(self, preds, targets, masks):  # predictions, targets, model
 
                 # Mask regression
                 if tuple(masks.shape[-2:]) != (mask_h, mask_w):  # downsample
-                    masks = F.interpolate(masks[None], (mask_h, mask_w), mode="bilinear", align_corners=False)[0]
+                    masks = F.interpolate(masks[None], (mask_h, mask_w), mode="nearest")[0]
                 marea = xywhn[i][:, 2:].prod(1)  # mask width, height normalized
                 mxyxy = xywh2xyxy(xywhn[i] * torch.tensor([mask_w, mask_h, mask_w, mask_h], device=self.device))
                 for bi in b.unique():