datasets/replica.py [42:58]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.rot_aug = rot_aug
        self.single_sample_per_trajectory = single_sample_per_trajectory
        self.samples_per_epoch = samples_per_epoch

        # step*seq_len < total_seq_len
        step = min(step, self.episode_len / seq_len)
        self.step = step
        self.resize_transform_rgb = transforms.Compose([transforms.Resize(self.img_res), transforms.ToTensor()])
        self.resize_transform_depth = transforms.Compose([transforms.Resize(self.img_res)])

    def get_trajectory_Rt(self):
        Rt = []
        for idxstr in self.seq_idxs:
            episode_Rt = []
            episode_path = os.path.join(self.datapath, idxstr)
            with open(os.path.join(episode_path, 'cameras.json'), 'r') as f:
                cameras = json.load(f)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



datasets/vizdoom.py [42:58]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.rot_aug = rot_aug
        self.single_sample_per_trajectory = single_sample_per_trajectory
        self.samples_per_epoch = samples_per_epoch

        # step*seq_len < total_seq_len
        step = min(step, self.episode_len / seq_len)
        self.step = step
        self.resize_transform_rgb = transforms.Compose([transforms.Resize(self.img_res), transforms.ToTensor()])
        self.resize_transform_depth = transforms.Compose([transforms.Resize(self.img_res)])

    def get_trajectory_Rt(self):
        Rt = []
        for idxstr in self.seq_idxs:
            episode_Rt = []
            episode_path = os.path.join(self.datapath, idxstr)
            with open(os.path.join(episode_path, 'cameras.json'), 'r') as f:
                cameras = json.load(f)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -