search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1k stars 333 forks source link

[Visualization] Adding an API to visualize segmentation maps along with images #1865

Closed ariG23498 closed 1 year ago

ariG23498 commented 1 year ago

Short Description

It would be great to include an API keras_cv.visualization.plot_segmentation_map_gallery() which would help with the visualition of the image along with their segmentation maps.

An exmaple of using this API would be: Segmentation maps

ariG23498 commented 1 year ago

I would like to take this issue up!

jbischof commented 1 year ago

@ariG23498 why does the segmentation map show up twice?

ariG23498 commented 1 year ago

@jbischof the API support y_true and y_pred. In this plot, I have the images, the ground truth masks and also the y_pred (which are the same masks).

This was just to demo the API -- I should have been more clear about this.