Failure in segmenting OD from DRIONS DB about optic-nerve-cnn HOT 1 OPEN

@Mahanteshambi, big apologies for not noticing your request in time.
Actually you just needed to add one line which rescales an image from uint8 [0, 255] scale to float [0, 1] scale of intensities:

im = np.array(Image.open(img_path))
im = im[0:, 40:]
print(im.shape)
im = cv2.resize(im, (256,256))

im = im.astype(np.float64) / 255.0

print(im.shape)
plt.imshow(im), plt.show()
im = np.expand_dims(im, axis=0)
im = tf_to_th_encoding(im)
prediction = (model.predict(im)[0, 0]).astype(np.float64)
plt.imshow(prediction, cmap=plt.cm.Greys_r), plt.show()

This way it works fine for me.

Also, it's necessary to add CLAHE im = skimage.exposure.equalize_adapthist(im) right after the rescaling line im = im.astype(np.float64) / 255.0 in order to reproduce exactly the same results, since model expects images only after CLAHE was performed on them.

from optic-nerve-cnn.