Coverage fix

microsoft · Jul 12, 2023 · 9263b52 · 9263b52
1 parent 2f8a82a
commit 9263b52
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/torchgeo/datamodules/vhr10.py b/torchgeo/datamodules/vhr10.py
@@ -17,8 +17,6 @@
 from ..transforms import AugmentationSequential
 from .geo import NonGeoDataModule
 
-# kornia DataKeys must accept MASKS
-
 
 class _AugPipe(Module):
     """Pipeline for applying augmentations sequentially on select data keys."""

diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py
@@ -39,10 +39,7 @@ def convert_coco_poly_to_mask(
         mask = torch.as_tensor(mask, dtype=torch.uint8)
         mask = mask.any(dim=2)
         masks.append(mask)
-    if masks:
-        masks_tensor = torch.stack(masks, dim=0)
-    else:
-        masks_tensor = torch.zeros((0, height, width), dtype=torch.uint8)
+    masks_tensor = torch.stack(masks, dim=0)
     return masks_tensor