Generating LMDB for Caffe

后端 未结 3 1061
走了就别回头了
走了就别回头了 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:28

    In caffe both lmdb and hdf5 supports multiple labels per image, matrices if you like, see this thread:

    https://github.com/BVLC/caffe/issues/1698#issue-53768814

    See this tutorial on how to create a multi-label dataset (lmdb here) for caffe with python code:

    http://www.kostyaev.me/article/Multilabel%20Dataset/

    EDIT: For example for the labels it uses the caffe-python function which converts a 3-dimensional array to datum, found in caffe/python/caffe.io.py: array_to_datum(arr, label=None):

提交回复
热议问题