From ba10066aff51f935f50ddba904ea34046433a79d Mon Sep 17 00:00:00 2001 From: Ross Girshick Date: Fri, 20 Dec 2013 14:39:22 -0800 Subject: [PATCH] demo on how to get net weights using the matlab interface --- matlab/caffe/matcaffe_demo_weights.m | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 matlab/caffe/matcaffe_demo_weights.m diff --git a/matlab/caffe/matcaffe_demo_weights.m b/matlab/caffe/matcaffe_demo_weights.m new file mode 100644 index 0000000..aae1559 --- /dev/null +++ b/matlab/caffe/matcaffe_demo_weights.m @@ -0,0 +1,35 @@ +function layers = matcaffe_demo_weights(use_gpu) +% layers = matcaffe_demo_weights(use_gpu) +% +% Demo of how to extract network parameters ("weights") using the matlab +% wrapper. +% +% input +% use_gpu 1 to use the GPU, 0 to use the CPU +% +% output +% layers struct array of layers and their weights +% +% You may need to do the following before you start matlab: +% $ export LD_LIBRARY_PATH=/opt/intel/mkl/lib/intel64:/usr/local/cuda/lib64 +% $ export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6 +% Or the equivalent based on where things are installed on your system + +% init caffe network (spews logging info) +if caffe('is_initialized') == 0 + model_def_file = '../../examples/imagenet_deploy.prototxt'; + model_file = '../../examples/alexnet_train_iter_470000'; + caffe('init', model_def_file, model_file); +end + +% set to use GPU or CPU +if exist('use_gpu', 'var') && use_gpu + caffe('set_mode_gpu'); +else + caffe('set_mode_cpu'); +end + +% put into test mode +caffe('set_phase_test'); + +layers = caffe('get_weights'); -- 2.7.4