protobuf、prototxt的使用/Windows/VS [操作系统入门]

Z时代
2024-01-10
分类：综合

博客已转移至个人网站（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 "gloglogging.h"
#include "acf.pb.h"
#include "io.hpp"
usingnamespace 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(constchar* 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, constchar* 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(constchar* 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, constchar* 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