Thank you very much Miranda. That feature is not yet supported, but you can obtain the array of the segmentation’s output by using this code

segmap, output = segment_image.segmentAsAde20k(path_to_image)