code: def postprocess(self, results):
      results = np.argmax(results, axis=1)
      results = results[np.newaxis, :, :, :, :]
      return results
  
imports: import numpy
  from medicalseg.cvlibs import manager
  from medicalseg.inference_helpers import InferenceHelper
label_desc: Use image node_id 2 for calling the InferenceHelper2D obj's underlying member method code with example usage: obj.postprocess(results) and returns: results