使用caffe进行语义分割训练的数据准备

语义分割任务的特点:

data:图片 彩色图:[N,3,W,H]  灰度图:[N,1,W,H] 

label:图片[N,C,W,H]  C为分割目标类别数

(1)使用python接口,使用type=Python的自定义层进行数据处理

(2)使用LMDB数据库的方式

分类问题:LMDB数据库结合Datum的方式,可以对data和label进行统一封装。

Datum.data  数据

Datum.label 标签

lmdb_txn.put(keystr,datum.SerializeToString())实现一个Datum向lmdb数据库的封装

分割问题:

使用两个LMDB分别存储data和label,在train_val.prototxt文件中定义两个Data层,后端连接对应的lmdb数据库,分别传入数据data和mask。

(2)使用HDF5 Layer实现

https://stackoverflow.com/questions/31774953/test-labels-for-regression-caffe-float-not-allowed/31808324#31808324

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值