Why is there NaN in the weights of Convolutional layer in the deeplab V3+ semantic segmentation network

3 次查看（过去 30 天）
Zhuofan Zheng
Zhuofan Zheng 2019-7-8
回答： Ganesh Regoti 2019-7-30
I use the example in https://ww2.mathworks.cn/help/vision/examples/semantic-segmentation-using-deep-learning.html
however, I want to use the deeplab to segmentate the resomte sensing images. but it has four channel so I change the first conv layer's size from 3 to 4.
It did works, but the accuracy did not increase. I stoped the training, finding that the weights in the first conv layer became NaN. The loss is not NaN, I feel so confused that why this happens?
I have tried several times, all this problem occured.

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

回答（1 个）

Ganesh Regoti
Ganesh Regoti 2019-7-30
Assuming that you are using resnet18 pretrained network, the model is trained with a layer size of 3 and it expects 3 channeled input.
If the size of convolution layer is changed then the model must be retrained to adjust the weights of the layers in the network.
Deeplab uses SoftMax which could be a reason for the output not having any zeros or Nans.

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by