diff --git a/datasets/noise_free_test_images/estimate_prnu.py b/datasets/noise_free_test_images/estimate_prnu.py index 5654686..8f4a69e 100644 --- a/datasets/noise_free_test_images/estimate_prnu.py +++ b/datasets/noise_free_test_images/estimate_prnu.py @@ -47,3 +47,6 @@ for imageName in os.listdir(datasetPath): imagePrnuEstimateNpArray = imageWithPrnuNpArray - denoise_tv_chambolle(imageWithPrnuNpArray, weight=0.2, channel_axis=-1) imagesPrnuEstimateNpArray += [imagePrnuEstimateNpArray] + +showImageWithMatplotlib(np.array(imagesPrnuEstimateNpArray).mean(axis = 0)) +