Guest User

PY script

a guest
Jun 21st, 2020
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.08 KB | None | 0 0
  1. from skimage import io as skio
  2. import caffe
  3. import numpy as np
  4.  
  5. caffe.set_mode_gpu()
  6. caffe.set_device(0)
  7.  
  8. param_fn = 'C:/Users/apple/models/aaa/_cityyyyyy_iter_200000.caffemodel'
  9. net_fn = 'C:/Users/apple/models/100d.prototxt'
  10.  
  11. net = caffe.Net(net_fn,param_fn,caffe.TEST)
  12. transformer = caffe.io.Transformer({'data': [1,3,227,227]})
  13. transformer.set_transpose('data', (2,0,1))
  14. transformer.set_channel_swap('data', (2,1,0))
  15.  
  16.  
  17.  
  18. def feed_back(src,dst):
  19.  
  20. net.forward()
  21.  
  22. src.data[:] = np.interp(dst.data,(dst.data.min(),dst.data.max()),(src.data.min(),src.data.max()))
  23.  
  24. net.backward()
  25.  
  26.  
  27. src = net.blobs['data']
  28. dst = net.blobs['conv1_a_t']
  29. feed_back(src,dst)
  30.  
  31. num_loops = 20
  32.  
  33. img = skio.imread('C:/users/apple/walk/aaaaaa.jpg')
  34.  
  35. src.data[:] = transformer.preprocess('data',img)
  36.  
  37. for loops in range(num_loops):
  38. feed_back(src,dst)
  39.  
  40. result = transformer.deprocess('data',net.blobs['conv1_a_t'].data)
  41.  
  42. norm_result = np.interp(result,(result.min(),result.max()),(0,255)).astype('uint8') #final image
  43.  
  44. skio.imsave('out.jpg',norm_result)
Advertisement
Add Comment
Please, Sign In to add comment