码迷,mamicode.com
首页 > Windows程序 > 详细

protobuf、prototxt的使用/Windows/VS

时间:2020-08-21 16:42:55      阅读:139      评论:0      收藏:0      [点我收藏+]

标签:_for   nts   htm   evel   extract   根目录   out   not   源文件   

博客已转移至个人网站(http://www.p-chao.com)

分为:

一、配置protobuf

二、使用prototxt

 

下载源码,编译出protoc.exe文件和libprotobuf.lib libprotoc.lib库文件

    注意版本一致,debug出来的给debug用...

 使用方法:

protoc.exe 用来根据 proto文件 生成对应的 头文件 和 信息源文件

    用法:    protoc --cpp_out=./ person.proto

                    注意命令行中的空格

 libproto ..lib在自己的程序中,配合生成的头和源文件

#pragma(lib,...)  ...

 然后还要注意,需要将源中extract_includes.bat提取的头文件置于工程根目录下,并设置包含

 技术图片技术图片

  

protobuf具体的生成方法见博客  http://www.2cto.com/kf/201401/270281.html  

不谢!

 关于配置文件,prototxt(caffe中的格式,事实上可以是任何文本文件,二进制格式或者文本格式都可以)

 读写prototxt的配置文件需要以下头,用到了google的glogs,自行下载编译,linux下直接使用apt-get 就可以

下面是windows的头(linux稍微修改下include),不多废话,上代码

 

#include <fcntl.h>
#include <google/protobuf/io/coded_stream.h>
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/text_format.h>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/highgui/highgui_c.h>
#include <opencv2/imgproc/imgproc.hpp>
#include <stdint.h>
#include <stdio.h>
 
#include <algorithm>
#include <fstream>  // NOLINT(readability/streams)
#include <string>
#include <vector>
 
#include <fcntl.h>
#include <Windows.h>
 
#ifdef USE_LEVELDB
#include <leveldb/db.h>
#endif
 
//#include "caffe/common.hpp"
#include "glog\logging.h"
#include "acf.pb.h"
#include "io.hpp"
 
 
using namespace std;
 
// port for Win32
#ifdef _MSC_VER
#define open _open
#define close _close
#endif
 
namespace acf {
 
using google::protobuf::io::FileInputStream;
using google::protobuf::io::FileOutputStream;
using google::protobuf::io::ZeroCopyInputStream;
using google::protobuf::io::CodedInputStream;
using google::protobuf::io::ZeroCopyOutputStream;
using google::protobuf::io::CodedOutputStream;
using google::protobuf::Message;
 
bool ReadProtoFromTextFile(const char* filename, Message* proto) {
  int fd = open(filename, O_RDONLY);
  CHECK_NE(fd, -1) << "File not found: " << filename;
  FileInputStream* input = new FileInputStream(fd);
  bool success = google::protobuf::TextFormat::Parse(input, proto);
  delete input;
  close(fd);
  return success;
}
 
void WriteProtoToTextFile(const Message& proto, const char* filename) {
  int fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
  FileOutputStream* output = new FileOutputStream(fd);
  CHECK(google::protobuf::TextFormat::Print(proto, output));
  delete output;
  close(fd);
}
 
bool ReadProtoFromBinaryFile(const char* filename, Message* proto) {
  int fd = open(filename, O_RDONLY | O_BINARY);
  CHECK_NE(fd, -1) << "File not found: " << filename;
  ZeroCopyInputStream* raw_input = new FileInputStream(fd);
  CodedInputStream* coded_input = new CodedInputStream(raw_input);
  coded_input->SetTotalBytesLimit(1073741824, 536870912);
 
  bool success = proto->ParseFromCodedStream(coded_input);
 
  delete coded_input;
  delete raw_input;
  close(fd);
  return success;
}
 
void WriteProtoToBinaryFile(const Message& proto, const char* filename) {
  fstream output(filename, ios::out | ios::trunc | ios::binary);
  CHECK(proto.SerializeToOstream(&output));
}
 
}  // namespace caffe

 


然后读取prototxt中的配置来初始化参数

 

Opt opt;
    string filename = "./default.prototxt";
    ReadProtoFromTextFileOrDie(filename, &opt);

 这里解释一下,proto文件只是帮助生成一个类、而prototxt文件中存储有配置信息

 例如本例中

proto文件

// global configuration of the afc
package acf;
 
message Opt{
    optional string name = 1;
    optional string posGtDir = 2;
    optional string posImgDir = 3;
    optional string negImgDir = 4;
    optional string posWinDir = 5;
    optional string negWinDir = 6;
    optional int32    nWeak = 7;
    optional int32    modelDs = 8;
    optional int32    modelDsPad = 9;
    optional int32    stride = 10;
}
 
// it‘s a size type
message Size{
    optional int32    height = 1;
    optional int32    weidth = 2;
}

 

prototxt文件

 

# configure text
#the name of saved detector
name: "models/AcfDetector"
#the director of the database
posGtDir: "database/posGt"
posImgDir: "database/positive"
negImgDir: "database/negative"
posWinDir: "1"
negWinDir: "1"
#the number of weak classifier
nWeak: 32
nWeak: 128
nWeak: 512
nWeak: 2048
#the size of windows
modelDs{
    height: 48
    weidth: 48
}
modelDs{
    height: 50
    weidth: 50
}
#the stride in image
stride: 4

 

Tips:注释可以使用 // 也可以使用 #

 文件也可以选择成二进制,效率会高许多,如果还不明白,参见google的指导和API文档。

 

protobuf、prototxt的使用/Windows/VS

标签:_for   nts   htm   evel   extract   根目录   out   not   源文件   

原文地址:https://www.cnblogs.com/leoking01/p/13527297.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!