p307800/fast-mri
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

kleine aanpassingen voor weergave

Browse Source
This commit is contained in:
Stefan
2022-03-23 17:00:22 +01:00
parent 01f458d0db
commit 82c285b1b0
7 changed files with 38 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/7.Visualize_saliency.py
View File

@ -1,7 +1,6 @@
import argparse
import numpy as np
import matplotlib.pyplot as plt
# import matplotlib.cm as cm
parser = argparse.ArgumentParser(
description='Calculate the froc metrics and store in froc_metrics.yml')
@ -19,6 +18,7 @@ EXPERIMENT = args.experiment
SALIENCY_DIR = f'./../train_output/{EXPERIMENT}_{series_}/saliency.npy'
IMAGES_DIR = f'./../train_output/{EXPERIMENT}_{series_}/images_list.npy'
SEGMENTATION_DIR = f'./../train_output/{EXPERIMENT}_{series_}/segmentations.npy'
SLIDE = 10
########## load saliency map ############
heatmap = np.load(SALIENCY_DIR)
@ -38,13 +38,13 @@ fig, axes = plt.subplots(2,len(SERIES))
print(np.shape(axes))
print(np.shape(heatmap))
print(np.shape(images_list))
max_value = np.amax(heatmap)
min_value = np.amin(heatmap)
max_value = np.amax(heatmap[:,:,SLIDE,:])
min_value = np.amin(heatmap[:,:,SLIDE,:])
for indx in range(len(SERIES)):
print(indx)
axes[0,indx].imshow(images_list[:,:,12,indx],cmap='gray')
im = axes[1,indx].imshow(np.squeeze(heatmap[:,:,12,indx]),vmin=min_value, vmax=max_value)
axes[0,indx].imshow(np.transpose(images_list[:,:,SLIDE,indx]),cmap='gray')
im = axes[1,indx].imshow(np.transpose(np.squeeze(heatmap[:,:,SLIDE,indx])),vmin=min_value, vmax=max_value)
axes[0,indx].set_title(SERIES[indx])
axes[0,indx].set_axis_off()
axes[1,indx].set_axis_off()