diff options
author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-03-19 20:50:13 -0700 |
---|---|---|
committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-03-19 20:50:13 -0700 |
commit | ec6f3b8466f68fc82f8b25dd048c26d3fd630784 (patch) | |
tree | e807217b6682a1d4663686e18cef4f75910e275e /matlab | |
parent | dfec9471ea3153cb63c88d0a8d6e326ce6989e34 (diff) | |
download | caffe-ec6f3b8466f68fc82f8b25dd048c26d3fd630784.tar.gz caffe-ec6f3b8466f68fc82f8b25dd048c26d3fd630784.tar.bz2 caffe-ec6f3b8466f68fc82f8b25dd048c26d3fd630784.zip |
minor cleanup in rcnn-finetuning -- rcnn feature computation tested at
this commit (in addition to all caffe unit tests passing)
Diffstat (limited to 'matlab')
-rw-r--r-- | matlab/caffe/matcaffe.cpp | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/matlab/caffe/matcaffe.cpp b/matlab/caffe/matcaffe.cpp index 524393a9..1913abd4 100644 --- a/matlab/caffe/matcaffe.cpp +++ b/matlab/caffe/matcaffe.cpp @@ -25,7 +25,7 @@ static int init_key = -2; // matlab uses RGB color channel order // images need to have the data mean subtracted // -// Data coming in from matlab needs to be in the order +// Data coming in from matlab needs to be in the order // [width, height, channels, images] // where width is the fastest dimension. // Here is the rough matlab for putting image data into the correct @@ -131,7 +131,7 @@ static mxArray* do_get_weights() { prev_layer_name = layer_names[i]; const mwSize dims[2] = {layer_blobs.size(), 1}; mx_layer_cells = mxCreateCellArray(2, dims); - mxSetField(mx_layers, mx_layer_index, "weights", mx_layer_cells); + mxSetField(mx_layers, mx_layer_index, "weights", mx_layer_cells); mxSetField(mx_layers, mx_layer_index, "layer_names", mxCreateString(layer_names[i].c_str())); mx_layer_index++; @@ -142,18 +142,19 @@ static mxArray* do_get_weights() { // where width is the fastest dimension mwSize dims[4] = {layer_blobs[j]->width(), layer_blobs[j]->height(), layer_blobs[j]->channels(), layer_blobs[j]->num()}; - mxArray* mx_weights = mxCreateNumericArray(4, dims, mxSINGLE_CLASS, mxREAL); + mxArray* mx_weights = mxCreateNumericArray(4, dims, mxSINGLE_CLASS, + mxREAL); mxSetCell(mx_layer_cells, j, mx_weights); float* weights_ptr = reinterpret_cast<float*>(mxGetPr(mx_weights)); -// mexPrintf("layer: %s (%d) blob: %d %d: (%d, %d, %d) %d\n", -// layer_names[i].c_str(), i, j, layer_blobs[j]->num(), -// layer_blobs[j]->height(), layer_blobs[j]->width(), -// layer_blobs[j]->channels(), layer_blobs[j]->count()); + // mexPrintf("layer: %s (%d) blob: %d %d: (%d, %d, %d) %d\n", + // layer_names[i].c_str(), i, j, layer_blobs[j]->num(), + // layer_blobs[j]->height(), layer_blobs[j]->width(), + // layer_blobs[j]->channels(), layer_blobs[j]->count()); switch (Caffe::mode()) { case Caffe::CPU: - memcpy(weights_ptr, layer_blobs[j]->cpu_data(), + memcpy(weights_ptr, layer_blobs[j]->cpu_data(), sizeof(float) * layer_blobs[j]->count()); break; case Caffe::GPU: @@ -219,6 +220,7 @@ static void init(MEX_ARGS) { mxFree(param_file); mxFree(model_file); + // NOLINT_NEXT_LINE(runtime/threadsafe_fn) init_key = rand(); if (nlhs == 1) { plhs[0] = mxCreateDoubleScalar(init_key); |