-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
fixing object_id_to_object_name as it was breaking the gazing when us… #76
base: main
Are you sure you want to change the base?
Conversation
…ing mask-rcnn when the misclassification occurs
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi Tushar,
Thank you so much for working on this PR. It looks nice. Please take a look at the comments, and make sure that we run pre-commit
to format the code for a PR. If you cannot find the pointer for pre-commit
, we can find some time to set it up. Thank you!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
some clean up comments
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
left the minor comments, overall it looks good!
Thank you for working on this @tusharsangam , I have left some comments. Let me know if they make sense. |
@KavitShah1998 @jimmytyyang Let me know if this looks good to create merge request |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! please rebase to main before merging!
from spot_rl.envs.gaze_env import construct_config_for_gaze | ||
from spot_rl.envs.nav_env import construct_config_for_nav | ||
from spot_rl.envs.place_env import construct_config_for_place | ||
from spot_rl.envs.skill_manager import SpotSkillManager | ||
from spot_rl.utils.utils import map_user_input_to_boolean | ||
|
||
if __name__ == "__main__": | ||
run_the_loop = True # Don't break the test loop | ||
is_in_position = False # is it before the receptacle | ||
# Construct all three configs | ||
nav_config, pick_config, place_config = ( | ||
construct_config_for_nav(), | ||
construct_config_for_gaze(), | ||
construct_config_for_place(), | ||
) | ||
# Modify any values in the config | ||
nav_config.SUCCESS_DISTANCE = 0.10 | ||
place_config.SUCCESS_DISTANCE = 0.10 | ||
pick_config.SUCCESS_DISTANCE = 0.10 | ||
pick_config.MAX_EPISODE_STEPS = 350 | ||
|
||
while run_the_loop: | ||
# Send the modified configs in SpotSkillManager | ||
spotskillmanager = SpotSkillManager(nav_config, pick_config, place_config) | ||
if not is_in_position: | ||
spotskillmanager.nav("pick_table_05_45") | ||
# Reset Arm before gazing starts | ||
spotskillmanager.get_env().reset_arm() | ||
spotskillmanager.gaze_controller.reset_env_and_policy("ball") | ||
pick_stats = spotskillmanager.pick("ball") | ||
# print(pick_stats) | ||
spotskillmanager.get_env().reset_arm() | ||
spotskillmanager.gaze_controller.reset_env_and_policy("ball") | ||
is_in_position = False | ||
# Navigate to Test Receotacle | ||
spotskillmanager.nav("pick_table_05_45") | ||
is_in_position = True | ||
|
||
run_the_loop = map_user_input_to_boolean( | ||
"Do you want to continue to next test or dock & exit ?" | ||
) | ||
if not run_the_loop: | ||
spotskillmanager.dock() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's move all experimental files in experiments/object_perception_manipulation
folder inside spot_rl_experiments/spot_rl/
Also add a read-me there.
…ing mask-rcnn when the misclassification occurs, also adding the gaze test file in tests folder