Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from skimage import io as skio
- import caffe
- import numpy as np
- caffe.set_mode_gpu()
- caffe.set_device(0)
- param_fn = 'C:/Users/apple/models/aaa/_cityyyyyy_iter_200000.caffemodel'
- net_fn = 'C:/Users/apple/models/100d.prototxt'
- net = caffe.Net(net_fn,param_fn,caffe.TEST)
- transformer = caffe.io.Transformer({'data': [1,3,227,227]})
- transformer.set_transpose('data', (2,0,1))
- transformer.set_channel_swap('data', (2,1,0))
- def feed_back(src,dst):
- net.forward()
- src.data[:] = np.interp(dst.data,(dst.data.min(),dst.data.max()),(src.data.min(),src.data.max()))
- net.backward()
- src = net.blobs['data']
- dst = net.blobs['conv1_a_t']
- feed_back(src,dst)
- num_loops = 20
- img = skio.imread('C:/users/apple/walk/aaaaaa.jpg')
- src.data[:] = transformer.preprocess('data',img)
- for loops in range(num_loops):
- feed_back(src,dst)
- result = transformer.deprocess('data',net.blobs['conv1_a_t'].data)
- norm_result = np.interp(result,(result.min(),result.max()),(0,255)).astype('uint8') #final image
- skio.imsave('out.jpg',norm_result)
Advertisement
Add Comment
Please, Sign In to add comment