How do I save the segmented images ?

[VivekMahesh]

Hi, thanks for the code. I am getting the mIoU result. I wanted to see the segmented image and save those it in a folder. Could you please help me with this ?

[davidtvs]

Glad you find the code helpful 😄

There's nothing implemented to do it. There's a predict function that you can easily adapt to save the images. You can also have a look at imshow_batch it takes a batch of images and labels, lays them out on a grid, and displays them; you can use matplotlib.pyplot.savefig to save instead of displaying.

[VivekMahesh]

Thank you so much.