语义分割任务的特点:
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实现