深度学习triplet loss 做人再识别(person-reid)
2017-02-21 14:33
281 查看
最近需要做这个课题,采用深度学习做,废话不多说,直接说一下我的思路以及遇到的问题,希望能有人交流一下(哭,效果一直不好,找不到原因)。
主要是参照cvpr的这篇论文Person Re-Identification by Multi-Channel Parts-Based CNN with Improved Triplet Loss Function.pdf(链接http://z-yt.net/tmp/cvpr2016/content/papers/8851b335.pdf)。按照论文中的说法,其最简单的用原始triplet loss和一个网络分支在VIPER数据库上就能达到top1 34%的正确率。可是臣妾经过一番折腾,表示做不到啊,相同的网络和loss我的top1在15%左右,差距简直不要太大。(我的batch设为30)
我做深度学习用的是caffe,tripletloss层参考的是 博客http://blog.csdn.net/tangwei2014/article/details/46812153中的写法,另外我在triplet loss层前面又加上了一个select层用于将anchor、positive、negative分成三个输出作为loss层的输入。则样本的组织形式为如下
anchor1
anchor2
positive1
positive2
negative1
negative2
此处拿batchsize为6来举例,则一共有两对triplet三元组,(anchor1,positive1,negative1)和(anchor2,positive2,negative2),他们是一一对应的在一个批次中,当然这在组织imag_data的输入list中得事先组织好三元组。另外修改过的地方还有data_transformer.cpp加入了可以随机crop不同宽高的代码。
这是caffe中修改过的代码的文件
http://download.csdn.net/download/gu_gu_/9759615,由于鄙人才疏学浅,又是第一次做深度学习,在训练和运用上都没有经验,还望高人能够指点一二,不胜感激。
主要是参照cvpr的这篇论文Person Re-Identification by Multi-Channel Parts-Based CNN with Improved Triplet Loss Function.pdf(链接http://z-yt.net/tmp/cvpr2016/content/papers/8851b335.pdf)。按照论文中的说法,其最简单的用原始triplet loss和一个网络分支在VIPER数据库上就能达到top1 34%的正确率。可是臣妾经过一番折腾,表示做不到啊,相同的网络和loss我的top1在15%左右,差距简直不要太大。(我的batch设为30)
我做深度学习用的是caffe,tripletloss层参考的是 博客http://blog.csdn.net/tangwei2014/article/details/46812153中的写法,另外我在triplet loss层前面又加上了一个select层用于将anchor、positive、negative分成三个输出作为loss层的输入。则样本的组织形式为如下
anchor1
anchor2
positive1
positive2
negative1
negative2
此处拿batchsize为6来举例,则一共有两对triplet三元组,(anchor1,positive1,negative1)和(anchor2,positive2,negative2),他们是一一对应的在一个批次中,当然这在组织imag_data的输入list中得事先组织好三元组。另外修改过的地方还有data_transformer.cpp加入了可以随机crop不同宽高的代码。
这是caffe中修改过的代码的文件
http://download.csdn.net/download/gu_gu_/9759615,由于鄙人才疏学浅,又是第一次做深度学习,在训练和运用上都没有经验,还望高人能够指点一二,不胜感激。
相关文章推荐
- 深度学习-使用cuda加速卷积神经网络-手写数字识别准确率99.7%
- 【王晓刚】深度学习在图像识别中的研究进展与展望
- 百度深度学习的图像识别进展
- [王晓刚]深度学习在图像识别中的研究进展与展望
- 【王晓刚】深度学习在图像识别中的研究进展与展望
- 深度学习是图像识别的新潮流,并行大样本训练自动找到问题特征
- [王晓刚]图像识别中的深度学习
- 图像识别中的深度学习
- 图像识别中的深度学习
- 图像识别中的深度学习【香港中文大学王晓刚】
- 【王晓刚】深度学习在图像识别中的研究进展与展望
- 深度学习在图像识别中的应用--学习笔记2
- 深度学习在图像识别中的应用--学习笔记1
- 图像识别中的深度学习
- 深度学习之Matlab 转C++在iOS上测试CNN手型识别
- 大数据深度学习下车辆厂牌型号识别
- 深度学习在图像识别中的发展进程与展望
- 深度学习在图像识别中的应用--学习笔记4
- 基于Triplet loss 函数训练人脸识别深度网络
- 深度学习-人脸识别