unknown database backend

最新推荐文章于 2023-03-21 11:49:53 发布

AllyLi0224

最新推荐文章于 2023-03-21 11:49:53 发布

阅读量2.6k

点赞数

分类专栏： caffe

本文链接：https://blog.csdn.net/Allyli0022/article/details/73920885

版权

caffe 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

#include "caffe/util/db.hpp"
#include "caffe/util/db_leveldb.hpp"
#include "caffe/util/db_lmdb.hpp"

#include <string>
#define USE_LMDB = 1; //找不到定义，就自己定义一下，即可解决问题 lmdb就可使用了
namespace caffe { namespace db {

DB* GetDB(DataParameter::DB backend) {
  switch (backend) {
#ifdef USE_LEVELDB
  case DataParameter_DB_LEVELDB:
    return new LevelDB();
#endif  // USE_LEVELDB
#ifdef USE_LMDB
  case DataParameter_DB_LMDB:
    return new LMDB();
#endif  // USE_LMDB
  default:
    LOG(FATAL) << "Unknown database backend";
    return NULL;
  }
}

DB* GetDB(const string& backend) {
#ifdef USE_LEVELDB
  if (backend == "leveldb") {
    return new LevelDB();
  }
#endif  // USE_LEVELDB
#ifdef USE_LMDB
  if (backend == "lmdb") {
    return new LMDB();
  }
#endif  // USE_LMDB
  LOG(FATAL) << "Unknown database backend";
  return NULL;
}

}  // namespace db
}  // namespace caffe