예제 #1
0
 def get_reconstruction(self):
     floor_color = [120, 90, 10]
     wall_color = [10, 90, 130]
     return sg.create_reconstruction([self.floor_points, self.wall_points],
                                     [floor_color, wall_color],
                                     self.cameras, self.shot_positions,
                                     self.shot_rotations)
예제 #2
0
    def get_reconstruction(self,
                           rotation_noise=0.0,
                           position_noise=0.0,
                           camera_noise=0.0):
        floor_color = [120, 90, 10]
        wall_color = [10, 90, 130]

        positions = self.shot_positions
        if position_noise != 0.0:
            for p in positions:
                sg.perturb_points(p, position_noise)
        rotations = self.shot_rotations
        if position_noise != 0.0:
            for r in rotations:
                sg.perturb_rotations(r, rotation_noise)
        cameras = self.cameras
        if camera_noise != 0.0:
            for c in cameras:
                c.focal *= 1 + camera_noise

        return sg.create_reconstruction(
            [self.floor_points, self.wall_points],
            [floor_color, wall_color],
            cameras,
            self.shot_ids,
            positions,
            rotations,
            self.rig_instances,
            self.instances_positions,
            self.instances_rotations,
            self.rig_models,
        )
예제 #3
0
 def get_reconstruction(self):
     floor_color = [120, 90, 10]
     wall_color = [10, 90, 130]
     return sg.create_reconstruction(
         [self.floor_points, self.wall_points],
         [floor_color, wall_color],
         self.cameras, self.shot_positions,
         self.shot_rotations)
예제 #4
0
    def get_reconstruction(self) -> types.Reconstruction:
        floor_color = [120, 90, 10]
        wall_color = [10, 90, 130]

        return sg.create_reconstruction(
            points=np.asarray([self.floor_points, self.wall_points]),
            colors=np.asarray([floor_color, wall_color]),
            cameras=self.cameras,
            shot_ids=self.shot_ids,
            rig_shots=self.rig_instances,
            rig_positions=self.instances_positions,
            rig_rotations=self.instances_rotations,
            rig_cameras=self.rig_cameras,
            reference=self.reference,
        )
예제 #5
0
    def get_reconstruction(self) -> types.Reconstruction:
        floor_color = [120, 90, 10]
        wall_color = [10, 90, 130]

        return sg.create_reconstruction(
            [self.floor_points, self.wall_points],  # pyre-fixme [6]
            [floor_color, wall_color],
            self.cameras,
            self.shot_ids,
            self.shot_positions,
            self.shot_rotations,
            self.rig_instances,
            self.instances_positions,
            self.instances_rotations,
            self.rig_cameras,
        )