Merge pull request #396 from HWiese1980/fix_tight_bbox_no_hands_nonet…

…ype_error fix: fix tight_bbox related error in MeshGraphormer
Fannovel16 · Jul 8, 2024 · fc5c5c4 · fc5c5c4
2 parents 3c1d9de + e2bd235
commit fc5c5c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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