Fix mask check in PatchPixelSampler (#1971)

* fix mask check in PatchPixelSampler * change torch.Tensor to Tensor
nerfstudio-project · May 22, 2023 · d8f2446 · d8f2446
1 parent 27e2c87
commit d8f2446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerfstudio/data/pixel_samplers.py b/nerfstudio/data/pixel_samplers.py
@@ -278,7 +278,7 @@ def sample_method(
         mask: Optional[Tensor] = None,
         device: Union[torch.device, str] = "cpu",
     ) -> Int[Tensor, "batch_size 3"]:
-        if mask:
+        if isinstance(mask, Tensor):
             # Note: if there is a mask, sampling reduces back to uniform sampling
             indices = super().sample_method(batch_size, num_images, image_height, image_width, mask=mask, device=device)
         else: