Esempio n. 1
0
 def _reset(self):
     self.total_frame = 0
     self.total_reward = 0
     self.has_collided = 0
     obs = CameraRobotEnv._reset(self)
     self._flag_reposition()
     return obs
Esempio n. 2
0
 def  _reset(self):
     self.total_frame = 0
     self.total_reward = 0
     self._randomize_target()
     self._flag_reposition()
     obs = CameraRobotEnv._reset(self)       ## Important: must come after flat_reposition
     return obs
Esempio n. 3
0
    def _reset(self):
        self.total_frame = 0
        self.total_reward = 0
        obs = CameraRobotEnv._reset(self)
        self._flag_reposition()

        [p.removeBody(i) for i in self.marker_ids]
        self.markers = []
        self.marker_ids = []
        [j.reset_joint_state(0, 0) for j in self.nonWheelJoints]
        return obs
Esempio n. 4
0
    def _reset(self):
        self.total_frame = 0
        self.total_reward = 0
        obs = CameraRobotEnv._reset(self)
        self._flag_reposition()

        # initial_pos:
        quat_orientation = p.getQuaternionFromEuler([0, 0, 3.14 / 2])
        base_position = [-14.3, 5, 1.2]
        base_position = [0, 0, 3.2]
        obj_file_name = 'TeaCup.urdf'

        self._ballUniqueId = p.loadURDF(
            obj_file_name,
            basePosition=base_position,
            baseOrientation=quat_orientation,
            flags=p.URDF_USE_MATERIAL_COLORS_FROM_MTL,
            globalScaling=0.25)

        return obs
Esempio n. 5
0
 def _reset(self):
     CameraRobotEnv._reset(self)
     for flagId in self.semantic_flagIds:
         p.changeVisualShape(flagId, -1, rgbaColor=[1, 0, 0, 1])
Esempio n. 6
0
 def _reset(self):
     obs = CameraRobotEnv._reset(self)
     return obs
Esempio n. 7
0
    def _reset(self):
        self.total_frame = 0
        self.total_reward = 0

        obs = CameraRobotEnv._reset(self)
        return obs