1 // This is a script to upgrade "V0" network prototxts to the new format.
3 // upgrade_net_proto_binary v0_net_proto_file_in net_proto_file_out
6 #include <fstream> // NOLINT(readability/streams)
7 #include <iostream> // NOLINT(readability/streams)
10 #include "caffe/caffe.hpp"
11 #include "caffe/util/io.hpp"
12 #include "caffe/util/upgrade_proto.hpp"
16 using namespace caffe; // NOLINT(build/namespaces)
18 int main(int argc, char** argv) {
19 FLAGS_alsologtostderr = 1; // Print output to stderr (while still logging)
20 ::google::InitGoogleLogging(argv[0]);
22 LOG(ERROR) << "Usage: "
23 << "upgrade_net_proto_binary v0_net_proto_file_in net_proto_file_out";
27 NetParameter net_param;
28 string input_filename(argv[1]);
29 if (!ReadProtoFromBinaryFile(input_filename, &net_param)) {
30 LOG(ERROR) << "Failed to parse input binary file as NetParameter: "
34 bool need_upgrade = NetNeedsUpgrade(net_param);
37 success = UpgradeNetAsNeeded(input_filename, &net_param);
39 LOG(ERROR) << "Encountered error(s) while upgrading prototxt; "
40 << "see details above.";
43 LOG(ERROR) << "File already in latest proto format: " << input_filename;
46 WriteProtoToBinaryFile(net_param, argv[2]);
48 LOG(INFO) << "Wrote upgraded NetParameter binary proto to " << argv[2];