Generating LMDB for Caffe

后端 未结 3 1052
走了就别回头了
走了就别回头了 2021-02-15 04:45

I am trying to build a deep learning model for Saliency analysis using caffe (I am using the python wrapper). But I am unable to understand how to generate the lmdb data structu

3条回答
  •  一生所求
    2021-02-15 05:42

    You can approach this problem in two ways:

    1. Using HDF5 data layer instead of LMDB. HDF5 is more flexible and can support labels the size of the image. You can see this answer for an example of constructing and using HDF5 input data layer.

    2. You can have two LMDB input layers: one for the image and one for the label. Note that when you build the LMDB you must not use the 'shuffle' option in order to have the images and their labels in sync.

    Update: I recently gave a more detailed answer here.

提交回复
热议问题