Fix mask bug (#1891)

This commit is contained in:
Yuantao Chen authored and GitHub committed 2023-05-09 09:32:05 -07:00
1 parent 91aec92653
commit f92275603a
1 file changed
+1 -1
@@ -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)