Github 项目推荐 | Nvidia 图片风格转换工具 —— FastPhotoStyle

FastPhotoStyle 是 Nvidia 的图片风格转换工具，其中包含了照片变为各种艺术风格的算法。通过给定的照片和风格，该工具可通过特定的算法何成为用户所需要的图片。代码详细的实现原理在 Nividia 的论文里有详细地说明，论文地址：

https://arxiv.org/abs/1802.06474。

Github 地址

https://github.com/NVIDIA/FastPhotoStyle

安装

只能在以下环境中测试代码

• OS: Ubuntu 16.04
• CUDA: 9.1
• Python 2 from Anaconda2
• pytorch 0.3.0
• 设置环境变量： export ANACONDA=PATH-TO-YOUR-ANACONDA-LIBRARY export CUDA_PATH=/usr/local/cuda export PATH=${ANACONDA}/bin:${CUDA_PATH}/bin:$PATH export LD_LIBRARY_PATH=${ANACONDA}/lib:${CUDA_PATH}/bin64:$LD_LIBRARY_PATH export C_INCLUDE_PATH=${CUDA_PATH}/include
• 安装需求 sudo apt-get install -y axel imagemagick (Only used for demo) conda install pytorch=0.3.0 torchvision cuda90 -y -c pytorch pip install scikit-umfpack pip install -U setuptools pip install cupy pip install pynvrtc