Dataset handling in Change Detection Projects (arc gis tile exports and saved predictions)

show_prediction_label.py

filename = "vector_strip2_256x256_over32/vector_strip2_8790.TIF"

file1 = "Fold_0/"+filename
file2 = "Gts/"+filename

from skimage import io
import numpy as np
import matplotlib.pyplot as plt

# The prediction
img = io.imread(file1)
arr = np.asarray(img)
tmp = arr.flatten()
print("Prediction image range:", np.min(tmp), "-", np.max(tmp))

plt.imshow(arr)
plt.show()

# The ground truth - ps: the direct exports from ArcGis have different values, hence the conversion afterwards
img = io.imread(file2)
arr = np.asarray(img)

arr[arr <= 0] = 0
arr[arr == 65535] = 0 # hi ArcGis ghosts
arr[arr != 0] = 1

tmp = arr.flatten()
print("GT image range:", np.min(tmp), "-", np.max(tmp))

plt.imshow(arr)
plt.show()