mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
Fix mask bug (#1891)
This commit is contained in:
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)
|
||||
|
||||
Reference in new issue
Block a user