How can I train a convolutional neural network for both classification and regression?

Question

Matthew Fall 2018-12-11

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/435217-how-can-i-train-a-convolutional-neural-network-for-both-classification-and-regression

评论： Matthew Fall 2020-6-17

I would like to use the same convolutional neural network to classify and perform regression on images. In other words, I would like to have shared input and hidden layers, but then branch off into a regression output layer and a classification output layer. How can I do this?

Part of this problem is that I have a lot of float-valued images stored as .mat files, so I would like to use their file names instead of storing all of my data in memory. Is it possible to use an image datastore with 2 labels for each image, or something like it?

As an example, I would like to train a convolutional neural network to classify digits and determine their rotation. MathWorks already has examples for the classification task and for the regression task. I would like to couple the two problems.