Update Demo_MCWNNM_ADMM2.m

csjunxu · Oct 10, 2019 · 929b400 · 929b400
1 parent b2f1553
commit 929b400
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Demo_MCWNNM_ADMM2.m b/Demo_MCWNNM_ADMM2.m
@@ -77,7 +77,7 @@
     % calculate the PSNR
     Par.PSNR(Par.Iter, Par.image)  =   csnr( im_out, Par.I, 0, 0 );
     Par.SSIM(Par.Iter, Par.image)      =  cal_ssim( im_out, Par.I, 0, 0 );
-    imname = sprintf([Par.method '_nSig' num2str(nSig(1)) num2str(nSig(2)) num2str(nSig(3)) '_' Par.model '_Oite' num2str(Par.Iter) '_Iite' num2str(Par.maxIter) '_rho' num2str(rho) '_mu' num2str(mu) '_lambda' num2str(lambda) '_' im_dir(i).name]);
+    imname = sprintf([Par.method '_nSig' num2str(nSig(1)) num2str(nSig(2)) num2str(nSig(3)) '_' Par.model '_Oite' num2str(Par.Iter) '_Iite' num2str(Par.maxIter) '_rho' num2str(Par.rho) '_mu' num2str(Par.mu) '_lambda' num2str(Par.lambda) '_' im_dir(i).name]);
     imwrite(im_out/255, imname);
     fprintf('%s : PSNR = %2.4f, SSIM = %2.4f \n',im_dir(i).name, Par.PSNR(Par.Iter, Par.image),Par.SSIM(Par.Iter, Par.image)     );
 end
@@ -88,5 +88,5 @@
 mSSIM=mean(SSIM,2);
 fprintf('The best PSNR result is at %d iteration. \n',idx);
 fprintf('The average PSNR = %2.4f, SSIM = %2.4f. \n', mPSNR(idx),mSSIM);
-name = sprintf([Par.method '_' Par.model '_nSig' num2str(nSig(1)) num2str(nSig(2)) num2str(nSig(3)) '_' Par.model '_Oite' num2str(Par.Iter) '_Iite' num2str(Par.maxIter) '_rho' num2str(rho) '_mu' num2str(mu) '_lambda' num2str(lambda) '.mat']);
-save(name,'nSig','PSNR','SSIM','mPSNR','mSSIM');
+name = sprintf([Par.method '_' Par.model '_nSig' num2str(nSig(1)) num2str(nSig(2)) num2str(nSig(3)) '_' Par.model '_Oite' num2str(Par.Iter) '_Iite' num2str(Par.maxIter) '_rho' num2str(Par.rho) '_mu' num2str(Par.mu) '_lambda' num2str(Par.lambda) '.mat']);
+save(name,'nSig','PSNR','SSIM','mPSNR','mSSIM');