caffe 实现多标签输入(multilabel、multitask)

博客介绍了如何在Caffe中实现多标签输入,以支持多任务学习和多标签分类。作者详细说明了通过修改Caffe源代码,包括convert_imageset、io.hpp、io.cpp、caffe.proto、data_layers.hpp等多个文件,实现DataLayer、ImageDataLayer、MemoryDataLayer的多标签支持。实验结果显示输入标签与预期一致。文章提到了其他可能的方法,并链接到参考资料和作者的GitHub项目。
摘要由CSDN通过智能技术生成

         caffe 本身并不支持 多类标的输入, 该框架主要用于解决图片分类的问题,而目前,两个重要的问题需要多标签的输入:多任务学习(multi-task)和多标签分类(multi-label),本文针对这两个问题,实现了多标签的输入

 

         目前,网上流行的多标签输入方法主要有以下四种:

         1. 最简单,使用mxnet,它本身支持了多标签分类的问题,因此也自带了多标签的输入

         2. 用HDF5 + Slice Layer的方法完成,这种方法实现上没有什么难度,但是当数据量很大时,HDF5的存储方式会产生数十倍于图片的硬盘消耗,而且生成的过程也非常缓慢,本人一开始就主要使用这种方法,往往事倍功半

         3. 用两个data的输入(两个LMDB),一个只输出图片,一个只输出标签,这种方法相对前面两种要难一些,不过应该效果也不错

         4. 直接修改caffe的网络使其满足多标签的输入,为了方便以后的实验,本人实现了这种方法

 

         方法描述:注意到caffe的大多数数据转换都是从./.build_release/tools/convert_imageset 这种方法开始的,因此,从convert_i

评论 38
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值