What is the best way to get weights from a layer in DNN after loading some model?

kkudryavtsev
1 ●1

There is a function to SetWeights, but I did not find any function to read the weights?

1 answer

Sort by » oldest newest most voted

berak
32993 ●7 ●81 ●312

have a look at the docs , each Layer has a blobs member , which contains weights & biases. like:

for (auto b: layer->blobs) {
     for (int d=0; d<b.dims; d++)
         cout << b.size[d] << " ";
}

Thanks a lot! I figured out yesterday a little bit different approach with

...
for n in range( len( layer.blobs ) ):
         blob_ = net.getParam(LayerId, n )

kkudryavtsev ( 2018-11-13 10:29:55 -0600 )edit

1 follower

Asked: 2018-11-12 14:12:19 -0600

Seen: 543 times

Last updated: Nov 13 '18