fix

megvii-research · Jun 5, 2019 · 93a8fae · 93a8fae
1 parent 98ceaee
commit 93a8fae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/06-isp/demosaicing_sRGB/train.py b/06-isp/demosaicing_sRGB/train.py
@@ -121,7 +121,7 @@ def main():
                         placeholders['is_training']: False,
                     }
                     restored_v = sess.run([restored],feed_dict = feed_dict)
-                    psnr_x = compare_psnr(image[0,:,:,::-1], restored_v[0][0, :, :, ::-1])
+                    psnr_x = compare_psnr(gt_image[0,:,:,::-1], restored_v[0][0, :, :, ::-1])
                     psnrs.append(psnr_x)
                 print('average psnr is {:2.2f} dB'.format(np.mean(psnrs)))
         print('Training is done, exit.')