Mercurial > repos > PredictPDPS / file diff

--- a/src/PredictPDPS.jl	Sun Apr 21 19:26:50 2024 +0300
+++ b/src/PredictPDPS.jl	Sun Apr 21 20:42:43 2024 +0300
@@ -40,6 +40,7 @@
 include("AlgorithmBothNL.jl")
 include("AlgorithmFB.jl")
 include("AlgorithmFBDual.jl")
+include("PlotResults.jl")
 
 
 # Additional
@@ -52,6 +53,7 @@
 include("AlgorithmZeroDual.jl")
 include("PET/PET.jl")
 
+
 import .Algorithm,
        .AlgorithmBoth,
        .AlgorithmBothGreedyV,
@@ -66,17 +68,12 @@
        .AlgorithmNoPrediction,
        .AlgorithmPrimalOnly,
        .AlgorithmDualScaling,
-       .AlgorithmZeroDual,
-       .PET
+       .AlgorithmZeroDual
 
 using .ImGenerate
 using .OpticalFlow: DisplacementFull, DisplacementConstant
-using .PET: demo_petS1, demo_petS2, demo_petS3, 
-            demo_petS4, demo_petS5, demo_petS6, demo_petS7,
-            demo_petB1, demo_petB2, demo_petB3, 
-            demo_petB4, demo_petB5, demo_petB6, demo_petB7,
-            batchrun_shepplogan, batchrun_brainphantom, batchrun_pet,
-            plot_pet
+using .PlotResults
+using .PET
 
 ##############
 # Our exports
@@ -95,7 +92,7 @@
        demo_petB1, demo_petB2, demo_petB3, 
        demo_petB4, demo_petB5, demo_petB6, demo_petB7,
        batchrun_shepplogan, batchrun_brainphantom, batchrun_pet,
-       plot_pet
+       plot_denoising, plot_pet
        
 ###################################
 # Parameterisation and experiments
@@ -590,6 +587,16 @@
     end
 end
 
+#########################
+# Plotting SSIM and PSNR
+#########################
+
+function plot_denoising(kwargs...)
+    ssim_plot("lighthouse")
+    psnr_plot("lighthouse")
+end
+
+
 ###############
 # Precompiling
 ###############