| 3766 | assert self.n_imgs is not None |
| 3767 | |
| 3768 | def _set_random(self): |
| 3769 | assert 'max_angle' in self.params |
| 3770 | assert 'min_r' in self.params |
| 3771 | assert 'max_r' in self.params |
| 3772 | |
| 3773 | self.cam_poses: T.List[T.List[np.ndarray]] = [ |
| 3774 | rigid_motion.generate_random_camera_poses( |
| 3775 | n=self.n_imgs, |
| 3776 | max_angle=self.params.get('max_angle'), |
| 3777 | min_r=self.params.get('min_r'), |
| 3778 | max_r=self.params.get('max_r'), |
| 3779 | center_direction_w=self.params.get('center_direction_w', None), |
| 3780 | local_max_angle=self.params.get('local_max_angle', 0), |
| 3781 | rand_r=self.params.get('rand_r', 0), |
| 3782 | origin_w=self.params.get('origin_w', None), |
| 3783 | rng=self.rng, |
| 3784 | method=self.params.get('method', 'random'), |
| 3785 | dtype=self.np_dtype, |
| 3786 | ) |
| 3787 | for _ in range(self.total) |
| 3788 | ] # list of list of H_c2w |
| 3789 | self.cam_poses = utils.to_dtype( |
| 3790 | utils.to_tensor(self.cam_poses), |
| 3791 | dtype=self.torch_dtype, |
| 3792 | ) |
| 3793 | |
| 3794 | def _set_circle(self): |
| 3795 | |