also add -1

beneisner · beneisner · commit e9985e526821 · 2024-05-23T13:26:32.000-04:00
diff --git a/src/rpad/rlbench_utils/placement_dataset.py b/src/rpad/rlbench_utils/placement_dataset.py
@@ -206,6 +206,7 @@ def filter_out_names(rgb, point_cloud, mask, handlemapping, names=BACKGROUND_NAM
         handlemapping[name] for name in names if name in handlemapping
     ]
     background_handles.append(65535)  # It's -1, cast as uint16.
+    background_handles.append(-1)
     background_indices = np.isin(mask, background_handles).reshape((-1))
 
     # Get the indices of the foreground.

Original file line number	Diff line number	Diff line change
`@@ -206,6 +206,7 @@ def filter_out_names(rgb, point_cloud, mask, handlemapping, names=BACKGROUND_NAM`
`206`	`206`	`handlemapping[name] for name in names if name in handlemapping`
`207`	`207`	`]`
`208`	`208`	`background_handles.append(65535) # It's -1, cast as uint16.`
	`209`	`+ background_handles.append(-1)`
`209`	`210`	`background_indices = np.isin(mask, background_handles).reshape((-1))`
`210`	`211`
`211`	`212`	`# Get the indices of the foreground.`