src/datasets.py [466:477]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                     DatasetKeyConstants.image_pose: self.poses[index][None, :],
                     DatasetKeyConstants.image_rotation: self.rotations[index][None, :],
                     DatasetKeyConstants.ray_directions: self.directions}

        sample_dict = {}

        for feature_idx in range(len(self.train_config.f_in)):
            f_in = self.train_config.f_in[feature_idx]

            # get output of prepare_batch, which is the input to later batch function call
            in_prepared_batch = f_in.prepare_batch(data_item, self.config)
            sample_dict[f"{DatasetKeyConstants.batch_input_dir}_{feature_idx}"] = in_prepared_batch
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/datasets.py [508:519]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                     DatasetKeyConstants.image_pose: self.poses[index][None, :],
                     DatasetKeyConstants.image_rotation: self.rotations[index][None, :],
                     DatasetKeyConstants.ray_directions: self.directions}

        sample_dict = {}

        for feature_idx in range(len(self.train_config.f_in)):
            f_in = self.train_config.f_in[feature_idx]

            # get output of prepare_batch, which is the input to later batch function call
            in_prepared_batch = f_in.prepare_batch(data_item, self.config)
            sample_dict[f"{DatasetKeyConstants.batch_input_dir}_{feature_idx}"] = in_prepared_batch
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -