如何在AWS Sagemaker中使用分段掩码中的标签

Question

bounty到期在7小时内。这个问题的答案有资格获得+ 50声望赏金。 Dre正在寻找来自信誉良好的来源的答案：

请让我知道是否需要更多代码。

我正在从细分蒙版中检索蒙版中表示的标签。

这是我正在通过AWS Sagemaker中的语义细分模型运行的图像。

用于进行预测和显示遮罩的代码。

from sagemaker.predictor import json_serializer, json_deserializer, RealTimePredictor
from sagemaker.content_types import CONTENT_TYPE_CSV, CONTENT_TYPE_JSON

%%time
ss_predict = sagemaker.RealTimePredictor(endpoint=ss_model.endpoint_name, 
                                     sagemaker_session=sess,
                                    content_type = 'image/jpeg',
                                    accept = 'image/png')

return_img = ss_predict.predict(img)

from PIL import Image
import numpy as np
import io

num_labels = 21
mask = np.array(Image.open(io.BytesIO(return_img)))
plt.imshow(mask, vmin=0, vmax=num_labels-1, cmap='jet')
plt.show()

此图像是创建的分割蒙版，它代表摩托车，其他所有东西都是背景。

[<< img src =“ https://image.soinside.com/eyJ1cmwiOiAiaHR0cHM6Ly9pLnN0YWNrLmltZ3VyLmNvbS82RmJWbi5wbmcifQ==” alt =“分段蒙版[2]”>

您可以从代码中看到，有21种可能的标签，其中2种用于面罩，一种用于摩托车，另一种用于背景。我现在想弄清楚的是，如何在21种可能的选项中打印在此遮罩中实际使用的标签？

[如果您需要任何进一步的信息，请告诉我们，我们将不胜感激。