图像增强 cnn

https://github.com/HuiZeng/Image-Adaptive-3DLUT

该文是香港理工大学张磊老师及其学生在图像增强领域的又一颠覆性成果。它将深度学习技术与传统3DLUT图像增强技术结合，得到了一种更灵活、更高效的图像增强技术。所提方法能够以1.66ms的速度对4K分辨率图像进行增强(硬件平台：Titan RTX GPU)。

paper: https://www4.comp.polyu.edu.hk/~cslzhang/paper/PAMI_LUT.pdf

code: https://github.com/HuiZeng/Image-Adaptive-3DLUT

需要编译：

trilinear_cpp

  

   

    

     

    
    

     

      from setuptools import setup
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if torch.cuda.is_available():
     
    
   

    

     

    
    

     

       print('Including CUDA code.')
     
    
   

    

     

    
    

     

       setup(
     
    
   

    

     

    
    

     

       name='trilinear',
     
    
   

    

     

    
    

     

       ext_modules=[
     
    
   

    

     

    
    

     

       CUDAExtension('trilinear', [
     
    
   

    

     

    
    

     
 'src/trilinear_cuda.cpp',
     
    
   

    

     

    
    

     
 'src/trilinear_kernel.cu',
     
    
   

    

     

    
    

     

       ])
     
    
   

    

     

    
    

     

       ],
     
    
   

    

     

    
    

     

       cmdclass={
     
    
   

    

     

    
    

     
 'build_ext': BuildExtension
     
    
   

    

     

    
    

     

       })
     
    
   

    

     

    
    

     

      else:
     
    
   

    

     

    
    

     

       print('NO CUDA is found. Fall back to CPU.')
     
    
   

    

     

    
    

     

       setup(name='trilinear',
     
    
   

    

     

    
    

     

       ext_modules=[CppExtension('trilinear', ['src/trilinear.cpp'])],
     
    
   

    

     

    
    

     

       cmdclass={'build_ext': BuildExtension})
     
    
  
 

set DISTUTILS_USE_SDK=1
set MSSdk=1

编译成功，调用 dll找不到，解决方法：

把目录：Lib\site-packages\torch\lib

下面的dll拷贝到pyd目录下面，可以调用了。

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/111938303