How to edit the denoisingImageDatastore source code so that it applies Poisson noise instead of Gaussian noise?

조회 수: 3 (최근 30일)
I have been training my network for image denoising, and I've achieved solid results with removing Gaussian noise from noisy images. However, a more real-world like noise is Poisson noise. I've read through the source code for denoisingImageDatastore and nowhere could I find a line of code where the imnoise function is applied to the extracted patches. You can view the source code here so you don't have to dig for it. I've tried many other solution but to no avail, and this is the last one I'm willing to try actually. All help is appreciated.

채택된 답변

Johanna Pingel
Johanna Pingel 2019년 2월 8일
I'm pretty sure that example only uses Gaussian noise, which may not be easy to change to Poisson. You may have to set up sample images yourself. Try following this example to set up the training image structure instead.
I'm pretty sure the key is to set up a randomPatchExtractionDatastore that will contain both your input and output images to the network.
Then you'll want to set up the network. I haven't looked into how the network is set up for denoising, so you'll want to see what the example is doing and perhaps copy that.
Let me know if I can help further.
  댓글 수: 5
Jonathan Foo
Jonathan Foo 2019년 3월 1일
Have you figured out a way to do this? I'm looking to do something similar, but with speckle noise instead!
Johanna Pingel
Johanna Pingel 2019년 3월 7일
You're going to have to train a network to do this if it's not Gaussian: check out my blog where I walk through the process (and share code)
https://blogs.mathworks.com/deep-learning/2019/02/21/image-to-image-regression/

댓글을 달려면 로그인하십시오.

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Image Data Workflows에 대해 자세히 알아보기

제품


릴리스

R2018b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by