FilesExpand file tree

 master

/

upgrade_net_proto_binary.cpp

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

46 lines (38 loc) · 1.3 KB

 master

/

upgrade_net_proto_binary.cpp

Copy path

Top

File metadata and controls

• Code
• Blame

46 lines (38 loc) · 1.3 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

40

41

42

43

44

45

46

// Copyright 2014 BVLC and contributors.

//

// This is a script to upgrade "V0" network prototxts to the new format.

// Usage:

// upgrade_net_proto_binary v0_net_proto_file_in net_proto_file_out

#include <cstring>

#include <iostream> // NOLINT(readability/streams)

#include <fstream> // NOLINT(readability/streams)

#include "caffe/caffe.hpp"

#include "caffe/util/io.hpp"

#include "caffe/util/upgrade_proto.hpp"

using std::ofstream;

using namespace caffe; // NOLINT(build/namespaces)

int main(int argc, char** argv) {

::google::InitGoogleLogging(argv[0]);

if (argc != 3) {

LOG(ERROR) << "Usage: "

<< "upgrade_net_proto_binary v0_net_proto_file_in net_proto_file_out";

return 1;

}

NetParameter net_param;

if (!ReadProtoFromBinaryFile(argv[1], &net_param)) {

LOG(ERROR) << "Failed to parse input binary file as NetParameter: "

<< argv[1];

return 2;

}

bool need_upgrade = NetNeedsUpgrade(net_param);

bool success = true;

if (need_upgrade) {

NetParameter v0_net_param(net_param);

success = UpgradeV0Net(v0_net_param, &net_param);

} else {

LOG(ERROR) << "File already in V1 proto format: " << argv[1];

}

WriteProtoToBinaryFile(net_param, argv[2]);

LOG(ERROR) << "Wrote upgraded NetParameter binary proto to " << argv[2];

return !success;

}