多标签caffe重新编译

说明:
Caffe自带的图像转LMDB接口只支持单label,对于多label的任务,可以使用HDF5的格式,也可以通过修改caffe代码来实现.本篇文章介绍怎么通过修改DataLayer来实现带Multilabel的LMDB格式数据输入的分类任务.
--- 本文参考自 Caffe实现多标签分类支持Multi-Label的lmdb数据格式输入


共修改5个文件:

  • $CAFFE_ROOT/src/caffe/proto/caffe.proto
  • $CAFFE_ROOT/src/caffe/layers/data_layer.cpp
  • $CAFFE_ROOT/src/caffe/util/io.cpp
  • $CAFFE_ROOT/include/caffe/util/io.hpp
    - $CAFFE_ROOT/tools/convert_imageset.cpp

    其中$CAFFE_ROOT为caffe根目录,即git clone 目录, 如图:

猜你喜欢

转载自www.cnblogs.com/geoffreyone/p/10730239.html
今日推荐