src/caffe/util/io.hpp

   1 // Copyright Yangqing Jia 2013
   2
   3 #ifndef CAFFE_UTIL_IO_H_
   4 #define CAFFE_UTIL_IO_H_
   5
   6 #include <google/protobuf/message.h>
   7
   8 #include <string>
   9
  10 #include "caffe/proto/caffe.pb.h"
  11
  12 using std::string;
  13
  14 namespace caffe {
  15
  16 void ReadImageToProto(const string& filename, BlobProto* proto);
  17 void WriteProtoToImage(const string& filename, const BlobProto& proto);
  18
  19 void ReadProtoFromTextFile(const char* filename,
  20     ::google::protobuf::Message* proto);
  21 inline void ReadProtoFromTextFile(const string& filename,
  22     ::google::protobuf::Message* proto) {
  23   ReadProtoFromTextFile(filename.c_str(), proto);
  24 }
  25
  26 }  // namespace caffe
  27
  28 #endif   // CAFFE_UTIL_IO_H_