diff --git a/social_mpc/controller.py b/social_mpc/controller.py index b5644b78e1b2f9b79b14504cad5cf775b050e6c8..dbcaa581e1fa588560ed5e0b426ece4f8b586bbf 100644 --- a/social_mpc/controller.py +++ b/social_mpc/controller.py @@ -228,11 +228,21 @@ class RobotController(): self.shared_goto_target_human_id[1] = 0 self.shared_goto_target_group_id[0] = False self.shared_goto_target_group_id[1] = 0 + self.controller_config.goto_target_id = -1 + self.controller_config.human_target_id = -1 + self.controller_config.goto_target_pose = np.zeros_like(self.controller_config.goto_target_pose) + self.controller_config.pan_target_id = -1 config = self.controller_config self.goto_goal = np.zeros(config.goto_target_dim) self.pan_goal = np.zeros(config.pan_target_dim) self.human_features = np.zeros(config.human_target_dim) self.last_human_features = np.zeros(config.human_target_dim) + self.human_target_id = None + self.goto_target_flag = False + self.pan_target_flag = False + self.pan_target_pose = None + self.set_mode() + self.set_weights() def on_subprocess_exit(self, name):