-
Notifications
You must be signed in to change notification settings - Fork 1
/
profile-caffemodel.cpp
39 lines (34 loc) · 1.16 KB
/
profile-caffemodel.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#include <caffe/caffe.hpp>
#include <google/protobuf/io/coded_stream.h>
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/text_format.h>
#include <algorithm>
#include <iosfwd>
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include <iostream>
#include "caffe/common.hpp"
#include "caffe/proto/caffe.pb.h"
#include "caffe/util/io.hpp"
using namespace caffe;
using namespace std;
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;
int main(int argc, char * argv[])
{
string sCaffeModel(argv[1]);
size_t fileNameIdx = sCaffeModel.find('.', 0);
string sTxtFileName = sCaffeModel.substr(0, fileNameIdx);
sTxtFileName += ".txt";
NetParameter proto;
ReadProtoFromBinaryFile(sCaffeModel, &proto);
WriteProtoToTextFile(proto, sTxtFileName);
return 0;
}