03: CNN for regression problems

[AlessandroPela]

I have a general question:
I am looking at the course chapter 03 and I see that every Computer vision example is related to classification problems.
What do I have to change if I am working with a Regression Problem?

[Perian-Yan]

Hi,
I guess you can just get rid of the softmax, treating the convolution layer as a normal nonlinear layer.

[Perian-Yan]

There are many references using CNN for regression. (I didn't go into details of them)
https://www.datatechnotes.com/2019/12/how-to-fit-regression-data-with-cnn.html
https://github.com/rsyamil/cnn-regression

Hope this helps

[vimaurya]

Thank you, this was helpful.

[mrdbourke]

Thank you for helping @Perian-Yan !