Advertisement
Guest User

Untitled

a guest
Dec 8th, 2016
60
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.45 KB | None | 0 0
  1. net = caffe.Net('imagenet.prototxt', 'imagenet.caffemodel', caffe.TEST)
  2.  
  3. conv_1_weights = old_net.params['conv1'][0].data
  4. conv_1_biases = old_net.params['conv1'][1].data
  5.  
  6. conv_av_weights = np.mean(conv_1_weights, axis=1)
  7.  
  8. new_net = caffe.Net('new_network.prototxt', 'imagenet.caffemodel', caffe.TEST)
  9.  
  10. new_net.params['conv1'][0].data[...] = conv_av_weights
  11. new_net.params['conv1'][1].data = conv_1_biases
  12.  
  13. new_net.save('newweights.caffemodel')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement