update output

Signed-off-by: binliu <[email protected]>

vista3d_pipeline.py

@@ -430,16 +430,17 @@ class VISTA3DPipeline(Pipeline):
 
     def postprocess(self, outputs, **kwargs):
         for key, value in kwargs.items():
+            if key not in self.POSTPROCESSING_EXTRA_ARGS:
+                logging.warning(f"Cannot set parameter {key} for postprocessing.")
             if (
                 key in self._postprocess_params
                 and value != self._postprocess_params[key]
-            ):
-                logging.warning(
-                    f"Please set the parameter {key} during initialization."
+            ) or (key not in self._postprocess_params):
+                self._postprocess_params.update(kwargs)
+                self.postprocessing_transforms = self._init_postprocessing_transforms(
+                    **self._postprocess_params
                 )
 
-            if key not in self.POSTPROCESSING_EXTRA_ARGS:
-                logging.warning(f"Cannot set parameter {key} for postprocessing.")
         outputs = self.postprocessing_transforms(decollate_batch(outputs))
         return outputs