From 67f6910fc77dab65f10b3d627089e0911c24ad65 Mon Sep 17 00:00:00 2001 From: James Thewlis Date: Mon, 17 Mar 2014 20:48:37 +0000 Subject: [PATCH] Prevent blob from being freed at end of if statement scope Was causing nonsense data to be read, tripping the blob size check in Blob::CopyFrom --- tools/dump_network.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp index 48804f2..386b8a3 100644 --- a/tools/dump_network.cpp +++ b/tools/dump_network.cpp @@ -44,10 +44,10 @@ int main(int argc, char** argv) { ReadProtoFromBinaryFile(argv[2], &trained_net_param); vector* > input_vec; + shared_ptr > input_blob(new Blob()); if (strcmp(argv[3], "none") != 0) { BlobProto input_blob_proto; ReadProtoFromBinaryFile(argv[3], &input_blob_proto); - shared_ptr > input_blob(new Blob()); input_blob->FromProto(input_blob_proto); input_vec.push_back(input_blob.get()); } -- 2.7.4