示例#1
0
 def _output(cls, im, guide, coeffs):
     with tf.device('/gpu:0'):
         out = bilateral_slice_apply(coeffs,
                                     guide,
                                     im,
                                     has_offset=True,
                                     name='slice')
     return out
示例#2
0
文件： models.py 项目： KeyKy/hdrnet 
 def _output(cls, im, guide, coeffs):
   with tf.device('/gpu:0'):
     out = bilateral_slice_apply(coeffs, guide, im, has_offset=True, name='slice')
   return out
示例#3
0
 def _output(cls, im, guide, coeffs):
   # Uncomment if build gpu version
   # with tf.device('/gpu:0'):
   out = bilateral_slice_apply(coeffs, guide, im, has_offset=True, name='slice')
   return out