src/features.py [421:436]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        return ret_dict

    def postprocess(self, inference_dict, idx):
        network_out = inference_dict[FeatureSetKeyConstants.network_output]

        ray_directions = inference_dict[FeatureSetKeyConstants.nerf_input_feature_ray_directions]
        z_vals = inference_dict[FeatureSetKeyConstants.nerf_input_feature_z_vals]

        rgb_map, disp_map, acc_map, weights, depth_map = nerf_raw2outputs(
            network_out.reshape(ray_directions.shape[0], z_vals.shape[1], -1), z_vals, ray_directions)

        inference_dict[FeatureSetKeyConstants.postprocessed_network_output] = rgb_map
        inference_dict[FeatureSetKeyConstants.nerf_weights_output] = weights

        inference_dict[FeatureSetKeyConstants.nerf_estimated_depth] = torch.reshape(
            self.depth_transform.from_world(depth_map, self.depth_range), (-1, 1))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/features.py [539:554]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        return ret_dict

    def postprocess(self, inference_dict, idx):
        network_out = inference_dict[FeatureSetKeyConstants.network_output]

        ray_directions = inference_dict[FeatureSetKeyConstants.nerf_input_feature_ray_directions]
        z_vals = inference_dict[FeatureSetKeyConstants.nerf_input_feature_z_vals]

        rgb_map, disp_map, acc_map, weights, depth_map = nerf_raw2outputs(
            network_out.reshape(ray_directions.shape[0], z_vals.shape[1], -1), z_vals, ray_directions)

        inference_dict[FeatureSetKeyConstants.postprocessed_network_output] = rgb_map
        inference_dict[FeatureSetKeyConstants.nerf_weights_output] = weights

        inference_dict[FeatureSetKeyConstants.nerf_estimated_depth] = torch.reshape(
            self.depth_transform.from_world(depth_map, self.depth_range), (-1, 1))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -