search

aws / amazon-sagemaker-examples

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.
https://sagemaker-examples.readthedocs.io
Apache License 2.0
9.98k stars 6.74k forks source link

How to access class labels and probabilities in Semantic Segmentation Inference #1196

Open OElesin opened 4 years ago

OElesin commented 4 years ago

I followed the SageMaker semantic segmentation example on GitHub, https://github.com/awslabs/amazon-sagemaker-examples/blob/master/introduction_to_amazon_algorithms/semantic_segmentation_pascalvoc/semantic_segmentation_pascalvoc.ipynb. However, in both inference cases the image mask is being plotted. I do not want the image masks. I need to access class labels and probabilities and show my application users classes and probabilities.

Currently, this is not explained in the tutorial. Any recommendations on how this might be solved?

yastasho commented 4 years ago

This question has been answered on https://forums.aws.amazon.com/thread.jspa?threadID=321586