您的位置:首页 > Web前端

改caffe 里面的contrsitve_loss

2015-12-04 16:00 246 查看
昨天按照如何在caffe中增加layer以及caffe中triplet loss layer的实现

链接地址为:/article/2280983.html

把triplet 网络放在caffe 里面。

今天把caffe里面的contrasitive_loss.cpp 由三个输入:fea1 fea2 sim 改成四个输入,

现在的输入为:fea1 fea2 label1 label2 ,我用代码比对软件对比了happynear 跟caffe 里里面的contraistive 层的不同,用happynear 中的contrastive_loss_layer.cpp 代替caffe 中的contrastive_loss_layer.cpp。

2. 改caffe 下面的include caffe layers 下面的contrsitve_loss_layer.hpp 下面 return 3 改为return 4

然后make all 没有报错。暂时认为改写成功了。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: