argmax_last_axis¶

argmax_last_axis(image)[source]¶

Define the post-processing of this class of model.

This simply applies argmax along last axis of the input.

Parameters:

image (np.ndarray) – The input image array.

Returns:

The post-processed image array.

Return type:

np.ndarray