| 3961 | self.cam_poses.append(poses) |
| 3962 | |
| 3963 | def _set_basic(self): |
| 3964 | assert 'min_r' in self.params |
| 3965 | assert 'max_r' in self.params |
| 3966 | max_r = self.params['max_r'] |
| 3967 | min_r = self.params['min_r'] |
| 3968 | |
| 3969 | # both input and target are around the same circle path |
| 3970 | self.cam_poses = [] |
| 3971 | for i in range(self.total): |
| 3972 | poses = [] |
| 3973 | |
| 3974 | Hs_c2w_input = utils.generate_camera_circle_path( |
| 3975 | num_poses=self.n_imgs, |
| 3976 | d_to_origin=-max_r / 2, |
| 3977 | r_circle=max_r, |
| 3978 | center_angles=[-90, 0], |
| 3979 | alt_yaxis=True, |
| 3980 | ) # (n, 4, 4) |
| 3981 | |
| 3982 | for j in range(Hs_c2w_input.size(0) - 1): # lfrb |
| 3983 | poses.append(Hs_c2w_input[j]) |
| 3984 | self.cam_poses.append(poses) |
| 3985 | |
| 3986 | def _set_grid(self): |
| 3987 | # build camera grid around one point |