Caffe精细调整以实现图像分割损失为零

时间:2018-12-07 21:45:48

标签: deep-learning caffe loss-function

我正在尝试接受训练有素的Imagenet分类网络,并对其进行微调以进行图像分割。我删除了完全连接的层,并用1x1卷积替换了它,以为每个类生成蒙版。我还用L1铰链损耗代替了损耗层。

在迭代0处,损耗约为10,000,然后在迭代20中,损耗变为0。当到达测试迭代时,其输出Test net output -nan。

有什么想法可能会出错吗?

0 个答案:

没有答案