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

最新推荐文章于 2021-02-21 17:54:45 发布

前进ing_嘟嘟

最新推荐文章于 2021-02-21 17:54:45 发布

阅读量643

点赞数

文章标签： caffe deep learning CustomData FlowNet

本文链接：https://blog.csdn.net/wendygelin/article/details/88396597

版权

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
  }
}