def _postprocess(self, results):
        "results is numpy array, optionally postprocess with argmax"
        if self.args.with_argmax:
            results = np.argmax(results, axis=1)
        return results
results is numpy array, optionally postprocess with argmax
contrib/MedicalSeg/deploy/python/infer.py