如何通过matconvnet将自定义图层和损失函数添加到预训练的CNN模型中?

时间:2017-04-06 08:26:25

标签: loss matconvnet

我是matconvnet的新手。最近,我想在预训练模型中尝试新的损失函数而不是现有的函数,例如vgg-16,它通常使用softmax损失层。更重要的是,我想使用新的特征提取器层,而不是池化层或最大层​​。我知道matconvnet,simpleNN和DagNN分别有2个CNN包装器,因为我使用的是vgg-16,一个具有线性序列构建块的线性模型。因此,在simpleNN包装器中,如何详细创建自定义层,特别是过程和相关概念,例如,我是否需要删除新功能提取器层后面的层或者只是留下它们?而且我知道如何计算损失函数的导数,因此层内的计算细节在这个问题中并不重要,我只想知道代码所代表的过程。有人能帮助我吗?我会非常感激!

1 个答案:

答案 0 :(得分:0)

您可以删除较旧的错误或目标图层 net.layer( 'ABC')= []; 并且您可以在vl_nnloss()文件中添加新的错误代码