diff --git a/node_wrappers/mesh_graphormer.py b/node_wrappers/mesh_graphormer.py index 5f5ddeb..1700f36 100644 --- a/node_wrappers/mesh_graphormer.py +++ b/node_wrappers/mesh_graphormer.py @@ -79,7 +79,7 @@ def execute(self, image, mask_bbox_padding=30, mask_type="based_on_depth", mask_ elif mask_type == "tight_bboxes": mask = np.zeros_like(mask) - hand_bboxes = info["abs_boxes"] + hand_bboxes = (info or {}).get("abs_boxes") or [] for hand_bbox in hand_bboxes: x_min, x_max, y_min, y_max = hand_bbox mask[y_min:y_max+1, x_min:x_max+1, :] = 255 #HWC