Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- net = caffe.Net('imagenet.prototxt', 'imagenet.caffemodel', caffe.TEST)
- conv_1_weights = old_net.params['conv1'][0].data
- conv_1_biases = old_net.params['conv1'][1].data
- conv_av_weights = np.mean(conv_1_weights, axis=1)
- new_net = caffe.Net('new_network.prototxt', 'imagenet.caffemodel', caffe.TEST)
- new_net.params['conv1'][0].data[...] = conv_av_weights
- new_net.params['conv1'][1].data = conv_1_biases
- new_net.save('newweights.caffemodel')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement