MATLAB Answers

Is there any "regression" output layer equivalent to pixelClass​ificationL​ayer?

I am working on a denoising Neural Network that has an image as input and outputs a denoised version of the image as output. I have created the network in Matlab, using the Deep Learning Toolbox, and now I need to train it. However, trainNetwork does not accept a Ground Truth that is not a categorical. I've doubled check my network and the issue might be that my last layer is a Classification Layer (pixelClassificationLayer). However, in my case, the problem is a regression one, where the value of each output pixel can correspond to any level of gray. I wondered, is there any regression equivalent to pixelClassificationLayer, or plausible alternative for Regression issues in Deep Learning Imaging?
Thank you very much.


Sign in to comment.

0 Answers