ho11laqe's picture
init
ecf08bc
raw
history blame
1.63 kB
import argparse
import os
from batchgenerators.utilities.file_and_folder_operations import join, isdir, maybe_mkdir_p
from skimage import io
import matplotlib.pyplot as plt
import matplotlib.image as pltimage
import cv2
def main(predictions, labels, sar_images):
prediction_files = os.listdir(predictions)
output_folder = join(predictions, 'eval')
maybe_mkdir_p(output_folder)
for file in prediction_files:
if file.endswith('.png'):
file = file[:-len('.png')]
prediction_path = join(predictions, file + '.png')
label_path = join(labels, file + '_front.png')
image_path = join(sar_images, file + '.png')
prediction = io.imread(prediction_path, as_gray=True)
label = io.imread(label_path)
image = io.imread(image_path)
image_rgb = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
image_rgb[label > 0] = [0, 255, 0]
image_rgb[prediction > 0] = [255, 0, 0]
output_path = join(output_folder, file+'.png')
pltimage.imsave(output_path, image_rgb)
print(file)
return
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("-p", '--predictions', help="Folder with png files of prediction")
parser.add_argument("-l", '--labels', help="Folder with png files of labels")
parser.add_argument("-i", '--sar_images', help="Folder with sar_images of glacier")
args = parser.parse_args()
predictions = args.predictions
labels = args.labels
sar_images = args.sar_images
main(predictions, labels, sar_images)