Loading vagen/env/primitive_skill/maniskill/env/align_two_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -45,12 +45,12 @@ class AlignTwoCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.6, 0.7, 0.6], [0.0, 0.0, 0.35]) return CameraConfig("render_camera", pose, 512, 512, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300,300, 1, 0.01, 100) def _load_scene(self, options: dict): self.cube_half_size = common.to_tensor([0.02] * 3) Loading vagen/env/primitive_skill/maniskill/env/place_two_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -46,12 +46,12 @@ class PlaceTwoCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.6, 0.7, 0.6], [0.0, 0.0, 0.35]) return CameraConfig("render_camera", pose, 512, 512, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300,300, 1, 0.01, 100) def _load_scene(self, options: dict): self.cube_half_size = common.to_tensor([0.02] * 3) Loading vagen/env/primitive_skill/maniskill/env/put_apple_in_drawer.py +2 −2 Original line number Diff line number Diff line Loading @@ -57,7 +57,7 @@ class PutAppleInDrawerEnv(BaseEnv): @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([1.0, 0.3, 1.0], [-0.5, -0.2, 0.35]) return CameraConfig("render_camera", pose, 2048, 2048, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300, 300, 1, 0.01, 100) def instruction(self): return "Please put the apple in the drawer and close the drawer." Loading Loading @@ -113,7 +113,7 @@ class PutAppleInDrawerEnv(BaseEnv): xyz = torch.zeros((b, 3), device=self.device) xyz[:, 2] = apple_height # Use apple's actual height sampler = randomization.UniformPlacementSampler(bounds=[ [self.workspace_x[0]+0.35, self.workspace_y[0]+0.5], [self.workspace_x[0]+0.35, self.workspace_y[0]+0.7], [self.workspace_x[1], self.workspace_y[1]] ], batch_size=b) apple_xy = sampler.sample(0.04, 100) # Adjust radius based on apple size if needed Loading vagen/env/primitive_skill/maniskill/env/stack_three_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -46,13 +46,13 @@ class StackThreeCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.8, 0.8, 0.6], [0.14, +0.08, 0.12]) return CameraConfig( "render_camera", pose=pose, width=2048, height=2048, fov=0.63, near=0.01, far=100 "render_camera", pose=pose, width=300, height=300, fov=0.63, near=0.01, far=100 ) def instruction(self): Loading Loading
vagen/env/primitive_skill/maniskill/env/align_two_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -45,12 +45,12 @@ class AlignTwoCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.6, 0.7, 0.6], [0.0, 0.0, 0.35]) return CameraConfig("render_camera", pose, 512, 512, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300,300, 1, 0.01, 100) def _load_scene(self, options: dict): self.cube_half_size = common.to_tensor([0.02] * 3) Loading
vagen/env/primitive_skill/maniskill/env/place_two_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -46,12 +46,12 @@ class PlaceTwoCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.6, 0.7, 0.6], [0.0, 0.0, 0.35]) return CameraConfig("render_camera", pose, 512, 512, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300,300, 1, 0.01, 100) def _load_scene(self, options: dict): self.cube_half_size = common.to_tensor([0.02] * 3) Loading
vagen/env/primitive_skill/maniskill/env/put_apple_in_drawer.py +2 −2 Original line number Diff line number Diff line Loading @@ -57,7 +57,7 @@ class PutAppleInDrawerEnv(BaseEnv): @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([1.0, 0.3, 1.0], [-0.5, -0.2, 0.35]) return CameraConfig("render_camera", pose, 2048, 2048, 1, 0.01, 100) return CameraConfig("render_camera", pose, 300, 300, 1, 0.01, 100) def instruction(self): return "Please put the apple in the drawer and close the drawer." Loading Loading @@ -113,7 +113,7 @@ class PutAppleInDrawerEnv(BaseEnv): xyz = torch.zeros((b, 3), device=self.device) xyz[:, 2] = apple_height # Use apple's actual height sampler = randomization.UniformPlacementSampler(bounds=[ [self.workspace_x[0]+0.35, self.workspace_y[0]+0.5], [self.workspace_x[0]+0.35, self.workspace_y[0]+0.7], [self.workspace_x[1], self.workspace_y[1]] ], batch_size=b) apple_xy = sampler.sample(0.04, 100) # Adjust radius based on apple size if needed Loading
vagen/env/primitive_skill/maniskill/env/stack_three_cube.py +2 −2 Original line number Diff line number Diff line Loading @@ -46,13 +46,13 @@ class StackThreeCubeEnv(BaseEnv): @property def _default_sensor_configs(self): pose = sapien_utils.look_at(eye=[0.3, 0, 0.6], target=[-0.1, 0, 0.1]) return [CameraConfig("base_camera", pose, 128, 128, np.pi / 2, 0.01, 100)] return [CameraConfig("base_camera", pose, 300, 300, np.pi / 2, 0.01, 100)] @property def _default_human_render_camera_configs(self): pose = sapien_utils.look_at([0.8, 0.8, 0.6], [0.14, +0.08, 0.12]) return CameraConfig( "render_camera", pose=pose, width=2048, height=2048, fov=0.63, near=0.01, far=100 "render_camera", pose=pose, width=300, height=300, fov=0.63, near=0.01, far=100 ) def instruction(self): Loading
