diff --git a/eg3d/training/triplane.py b/eg3d/training/triplane.py
index 86e70d88..2ed921ed 100644
--- a/eg3d/training/triplane.py
+++ b/eg3d/training/triplane.py
@@ -86,7 +86,7 @@ def synthesis(self, ws, c, neural_rendering_resolution=None, update_emas=False,
         rgb_image = feature_image[:, :3]
         sr_image = self.superresolution(rgb_image, feature_image, ws, noise_mode=self.rendering_kwargs['superresolution_noise_mode'], **{k:synthesis_kwargs[k] for k in synthesis_kwargs.keys() if k != 'noise_mode'})
 
-        return {'image': sr_image, 'image_raw': rgb_image, 'image_depth': depth_image, 'triplane': planes}
+        return {'image': sr_image, 'image_raw': rgb_image, 'image_depth': depth_image,'image_feature':feature_image, 'triplane': planes}
     
     def sample(self, coordinates, directions, z, c, truncation_psi=1, truncation_cutoff=None, update_emas=False, **synthesis_kwargs):
         # Compute RGB features, density for arbitrary 3D coordinates. Mostly used for extracting shapes.