| 3829 | self.cam_poses.append(poses) # list of list of H_c2w |
| 3830 | |
| 3831 | def _set_udlrfb(self): |
| 3832 | # fixed 6 input views: up down left right front back |
| 3833 | assert 'min_r' in self.params |
| 3834 | assert 'max_r' in self.params |
| 3835 | max_r = self.params['max_r'] |
| 3836 | min_r = self.params['min_r'] |
| 3837 | |
| 3838 | assert self.n_imgs == 6 |
| 3839 | self.cam_poses = [] # (total, n_imgs, 4, 4) |
| 3840 | for i in range(self.total): |
| 3841 | r = self.rng.rand(1) * (max_r - min_r) + min_r |
| 3842 | poses = [] |
| 3843 | |
| 3844 | Hs_c2w_ud = utils.generate_camera_circle_path( |
| 3845 | num_poses=3, |
| 3846 | d_to_origin=0, |
| 3847 | r_circle=r, |
| 3848 | center_angles=[0, 0], |
| 3849 | alt_yaxis=True, |
| 3850 | ) # (n, 4, 4) |
| 3851 | Hs_c2w_lrfb = utils.generate_camera_circle_path( |
| 3852 | num_poses=5, |
| 3853 | d_to_origin=0, |
| 3854 | r_circle=r, |
| 3855 | center_angles=[0, 90], |
| 3856 | alt_yaxis=True, |
| 3857 | ) # (n, 4, 4) |
| 3858 | poses.append(Hs_c2w_ud[0]) # u |
| 3859 | for j in range(Hs_c2w_lrfb.size(0) - 1): # lfrb |
| 3860 | poses.append(Hs_c2w_lrfb[j]) |
| 3861 | poses.append(Hs_c2w_ud[1]) # d |
| 3862 | self.cam_poses.append(poses) |
| 3863 | |
| 3864 | def _set_spiral(self): |
| 3865 | assert 'min_r' in self.params |