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=$/bin:$/bin:$PATH

export LD_LIBRARY_PATH=$/lib:$/bin64:$LD_LIBRARY_PATH

export C_INCLUDE_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

春節 AI 學習狂歡，精品課程 豪華特輯

優惠折上折，福利搶不停！

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！