save_tiff_imagej_compatible('pred_train.tif', pred_train, axes='YX')

[farukalamai]

When using this it's not saving the actual predicted image. I also wanted to know how can I save the predicted image in different format like jpg/png etc.

I have tried different method, nothing is working. Its' saving a blank image.

[farukalamai]

@jdeschamps

[tibuch]

Did you verify that you image is not empty?

A quick google search reveals how to save images as jpg e.g.:

from PIL import Image
im = Image.fromarray(arr)
im.save("your_file.jpeg")

[farukalamai]

I have tried this. It's just saving a blank image.