FlowNet 中caffe自定义层解读 —— type: "CustomData"

CustomData:

主要功能:读取数据库中的LMDB类型数据,并将其随机排布后存入top
数据准备:lmdb的格式为“image0的名字 image1的名字 flow的名字”,换行后下一组“image0的名字 image1的名字 flow的名字”等等

各参数解读,见下面代码的注释:

layer {
  name: "CustomData1"
  type: "CustomData"
  top: "blob0"   # lmdb中的Image 0
  top: "blob1"   # lmdb中的Image 1
  top: "blob2"   # lmdb中的Flow ground truth
  top: "unused1" # Occlusions, remove if not present
  include {
    phase: TRAIN #应用在训练环节
  }
  data_param {
    source: "/data/FlyingChairs_release_TRAIN_lmdb"
    batch_size: 8
    backend: LMDB
    preselection_label: 1
    rand_permute: true
    rand_permute_seed: 77 #此处将数据随机排布用的,暂时不用管
    slice_point: 3
    slice_point: 6
    slice_point: 8 
	# slice_point的意思是,将这几处作为断点,比如这个例子中,
	# 第0,1,2通道为image 0,第3,4,5通道为image 1,第6,7通道为flow,第8通道为“unused1”
    encoding: UINT8 # image 0 格式
    encoding: UINT8 # image 1 格式
    encoding: UINT16FLOW # flow 格式
    encoding: BOOL1 # “unused1”格式
    verbose: true
  }
}

猜你喜欢

转载自blog.csdn.net/wendygelin/article/details/88396597