Change the return value of synthesis(): now it additionally returns t…

…he feature image and triplane.

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.