Skip to content

Commit

Permalink
added conditional topic and crop method back for testing.
Browse files Browse the repository at this point in the history
  • Loading branch information
Benteng Ma committed Jun 21, 2024
1 parent 901f2c1 commit 5954f51
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions skills/src/lasr_skills/describe_people.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
from numpy2message import numpy2message
from .vision import GetCroppedImage, ImageMsgToCv2
import numpy as np
from validate_keypoints import ValidateKeypoints


class DescribePeople(smach.StateMachine):
Expand All @@ -38,7 +39,7 @@ def __init__(self):
)
smach.StateMachine.add(
"GET_IMAGE",
GetCroppedImage(object_name="person", crop_method="closest"),
GetCroppedImage(object_name="person", crop_method=crop_method, rgb_topic=rgb_topic),
transitions={
"succeeded": "CONVERT_IMAGE",
"failed": "SAY_GET_IMAGE_AGAIN",
Expand All @@ -57,7 +58,7 @@ def __init__(self):
)
smach.StateMachine.add(
"GET_IMAGE_AGAIN",
GetCroppedImage(object_name="person", crop_method="closest"),
GetCroppedImage(object_name="person", crop_method=crop_method, rgb_topic=rgb_topic),
transitions={"succeeded": "CONVERT_IMAGE", "failed": "SAY_CONTINUE"},
)

Expand Down

0 comments on commit 5954f51

Please sign in to comment.