(self, x)
return class_id, prob
postprocess

        Postprocess to the inference engine output.

        Args:
            x: Inference engine output.

        Returns: Output data after argmax.
postprocess
postprocess
def postprocess(self, x):
        """postprocess

        Postprocess to the inference engine output.

        Args:
            x: Inference engine output.

        Returns: Output data after argmax.
        """
        x = x.flatten()
        class_id = x.argmax()
        prob = x[class_id]
        return class_id, prob
def postprocess ( self , x ) : x = x . flatten ( ) class_id = x . argmax ( ) prob = x [ class_id ] return class_id , prob