From f92275603a7a8f7fd2f51b1e2acf2272ae2bdd37 Mon Sep 17 00:00:00 2001 From: Yuantao Chen <44546388+Tao-11-chen@users.noreply.github.com> Date: Wed, 10 May 2023 00:32:05 +0800 Subject: [PATCH] Fix mask bug (#1891) --- nerfstudio/data/datamanagers/base_datamanager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/data/datamanagers/base_datamanager.py b/nerfstudio/data/datamanagers/base_datamanager.py index 0167ff7fb..beb41cf39 100644 --- a/nerfstudio/data/datamanagers/base_datamanager.py +++ b/nerfstudio/data/datamanagers/base_datamanager.py @@ -89,7 +89,7 @@ def variable_res_collate(batch: List[Dict]) -> Dict: image = data.pop("image") mask = data.pop("mask", None) images.append(image) - if mask: + if mask is not None: masks.append(mask) new_batch: dict = nerfstudio_collate(batch)