plots update (lighter blue)

Author: kaftanski

constants.py

@@ -16,4 +16,4 @@
 
 CORR_FISSURE_LABEL_DEFAULT_TS = "/share/data_rechenknecht03_2/students/kaftan/FissureSegmentation/fissure-segmentation/results/corresponding_points_ts/fissures/simple/labels.npz"
 CLASSES = {1: 'LOF', 2: 'ROF', 3: 'RHF'}
-CLASS_COLORS = ('r', matplotlib.cm.get_cmap('Set1').colors[2], 'b')
+CLASS_COLORS = ('r', matplotlib.cm.get_cmap('Set1').colors[2], 'dodgerblue')

keypoint_plots.py

@@ -40,6 +40,12 @@ def plot_keypoints(image, mode, patid, slice_num, slice_dim=1, spacing=1.5, crop
     ax.scatter(pts_slice[:, pts_index[0]], pts_slice[:, pts_index[1]], marker='+', s=150, c='orangered', linewidths=2)
     save_fig(fig, result_folder, f'{mode}_keypoints_{patid}_slice{slice_num}', pdf=False)
 
+    fig = plt.figure(figsize=textwidth_to_figsize(2, aspect=img_slice.shape[0] / img_slice.shape[1]))
+    ax = fig.gca()
+    ax.set_axis_off()
+    ax.imshow(img_slice, cmap='gray')
+    save_fig(fig, result_folder, f'{patid}_slice{slice_num}', pdf=False)
+
 
 def plot_all(patid='s0070'):
     ds = ImageDataset(IMG_DIR_TS)
@@ -127,6 +133,6 @@ def plot_enhancement(patid='s0070'):
 
 if __name__ == '__main__':
     KP_MODES.remove('noisy')
-    # plot_all()
-    # plot_enhancement()
-    plot_bad_example_enhancement()
+    plot_all()
+    # # plot_enhancement()
+    # plot_bad_example_enhancement()

qualitative.py

@@ -159,5 +159,5 @@ def comparative_figures(patid='s0070'):
 
 
 if __name__ == '__main__':
-    # comparative_figures()
     kp_comparison_figure(ae=True)
+    comparative_figures()