def _set_pixel_scores(self, pixel_scores):
        "get softmaxed properly shaped pixel scores"
        tmp = np.transpose(pixel_scores, (0, 2, 3, 1))
        return self._softmax(tmp, axis=-1)[0, :, :, 1]
get softmaxed properly shaped pixel scores
serverless/openvino/omz/intel/text-detection-0004/nuclio/model_handler.py