diff --git a/src/caffe/util/io.hpp b/src/caffe/util/io.hpp
index 29f7f4173845b5b9e39ef21cc767ee0221b15357..57beef1dc00685f06294ff1e813b70f1cbc075f5 100644 (file)
--- a/src/caffe/util/io.hpp
+++ b/src/caffe/util/io.hpp
#include "caffe/proto/caffe.pb.h"
using std::string;
+using ::google::protobuf::Message;
namespace caffe {
}
void ReadProtoFromTextFile(const char* filename,
- ::google::protobuf::Message* proto);
+ Message* proto);
inline void ReadProtoFromTextFile(const string& filename,
- ::google::protobuf::Message* proto) {
+ Message* proto) {
ReadProtoFromTextFile(filename.c_str(), proto);
}
+void WriteProtoToTextFile(const Message& proto, const char* filename);
+inline void WriteProtoToTextFile(const Message& proto, const string& filename) {
+ WriteProtoToTextFile(proto, filename.c_str());
+}
+
+void ReadProtoFromBinaryFile(const char* filename,
+ Message* proto);
+inline void ReadProtoFromBinaryFile(const string& filename,
+ Message* proto) {
+ ReadProtoFromBinaryFile(filename.c_str(), proto);
+}
+
+void WriteProtoToBinaryFile(const Message& proto, const char* filename);
+inline void WriteProtoToBinaryFile(const Message& proto, const string& filename) {
+ WriteProtoToBinaryFile(proto, filename.c_str());
+}
+
+
} // namespace caffe
#endif // CAFFE_UTIL_IO_H_