Is there any "regression" output layer equivalent to pixelClassificationLayer?

Question

Virginia Fernandez Gonzalez 2019-6-7

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/466075-is-there-any-regression-output-layer-equivalent-to-pixelclassificationlayer

评论： Sai Bhargav Avula 2020-8-10

Hi,

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.

Virginia

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Binu 2019-10-29

There is a regression layer where you basically swap the last layer from classification to regression. Also there are transfer learning methods to convert from a classification to regression . However I have not tried this on a pixelLevel problems.

https://www.mathworks.com/help/deeplearning/examples/convert-classification-network-into-regression-network.html

Have you considered using your gray "levels" as a classes or bins and treat them as a classification problem?

请先登录，再进行评论。

请先登录，再回答此问题。