runnable version for full demo

LASR-at-Home · Feb 23, 2024 · b91b736 · b91b736
1 parent af28931
commit b91b736
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py b/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py
@@ -259,7 +259,7 @@ def predict_frame(head_frame, torso_frame, full_frame, head_mask, torso_mask,):
     head_frame = pad_image_to_even_dims(head_frame)
     torso_frame = pad_image_to_even_dims(torso_frame)
 
-    rst = ImageOfPerson.from_parent_instance(p.predict(full_frame))
+    rst = ImageOfPerson.from_parent_instance(p.predict(head_frame))
 
     return rst.describe()
 

diff --git a/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py b/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py
@@ -130,4 +130,4 @@ def describe(self) -> str:
                 wearables.append('a necktie')
             description += ", ".join(wearables[:-2] + [" and ".join(wearables[-2:])]) + '. '
 
-        return description
+        return description if description != "This customer has " else ""
diff --git a/skills/src/lasr_skills/describe_people.py b/skills/src/lasr_skills/describe_people.py
@@ -297,4 +297,4 @@ def execute(self, userdata):
             userdata['people'] = people
             # except Exception:
             #     return 'failed'
-            # return 'succeeded'
+            return 'succeeded'