ホーム>プログラミング関連>その他のソースコード
ダウンロード

waifu2x ncnn vulkan

waifu2xコンバーターのNCNN実装。 Vulkan APIを使用して、Intel / AMD / Nvidia / Apple-Siliconで速く実行されます。

WAIFU2X-NCNN-VULKANは、NCNNプロジェクトをユニバーサルニューラルネットワーク推論フレームワークとして使用しています。

ダウンロード

Intel/AMD/NVIDIA GPUのWindows/Linux/MacOS実行可能ファイルをダウンロードします

https://github.com/nihui/waifu2x-ncnn-vulkan/releases

このパッケージには、必要なすべてのバイナリとモデルが含まれています。ポータブルなので、cudaやcaffeのランタイム環境は必要ありません:)

使用法

例コマンド

waifu2x-ncnn-vulkan.exe -i input.jpg -o output.png -n 2 -s 2

完全な使用

 Usage: waifu2x-ncnn-vulkan -i infile -o outfile [options]...

  -h                   show this help
  -v                   verbose output
  -i input-path        input image path (jpg/png/webp) or directory
  -o output-path       output image path (jpg/png/webp) or directory
  -n noise-level       denoise level (-1/0/1/2/3, default=0)
  -s scale             upscale ratio (1/2/4/8/16/32, default=2)
  -t tile-size         tile size (>=32/0=auto, default=0) can be 0,0,0 for multi-gpu
  -m model-path        waifu2x model path (default=models-cunet)
  -g gpu-id            gpu device to use (-1=cpu, default=auto) can be 0,1,2 for multi-gpu
  -j load:proc:save    thread count for load/proc/save (default=1:2:2) can be 1:2,2,2:2 for multi-gpu
  -x                   enable tta mode
  -f format            output image format (jpg/png/webp, default=ext/png)
  • input-pathとoutput-pathファイルパスまたはディレクトリパスのいずれかを受け入れます
  • noise-level =ノイズレベル、大きな値は強いデノイズ効果を意味し、-1 =効果なし
  • scale =スケールレベル、1 =スケーリングなし、2 =アップスケール2x
  • tile-size =タイルサイズ、より小さな値を使用してGPUメモリの使用量を減らし、デフォルトは自動的に選択します
  • load:proc:save = 3つのステージ（画像デコード + waifu2x upscaling +画像エンコード）のスレッドカウント、より大きな値を使用すると、GPUの使用量が増加し、より多くのGPUメモリを消費する場合があります。この構成は、多くの小型画像で「4：4：4」で、大規模画像で「2：2：2」でチューニングできます。通常、デフォルトの設定は、ほとんどの状況では正常に動作します。 GPUがお腹が空いていることがわかった場合は、スレッドカウントを増やして処理を高速化してみてください。
  • format =出力する画像の形式、PNGはよりよくサポートされますが、WebPは一般に小さいファイルサイズを生成し、どちらも損失がないエンコードされています

クラッシュやエラーが発生した場合は、GPUドライバーをアップグレードしてみてください。

  • Intel：https：//downloadcenter.intel.com/product/80939/graphics-drivers
  • AMD：https：//www.amd.com/en/support
  • nvidia：https：//www.nvidia.com/download/index.aspx

ソースからビルド

  1. https://vulkan.lunarg.com/からVulkan SDKをダウンロードしてセットアップします
  • Linuxディストリビューションの場合、パッケージマネージャーから必須のビルド要件を取得できます
dnf install vulkan-headers vulkan-loader-devel
apt-get install libvulkan-dev
pacman -S vulkan-headers vulkan-icd-loader
  1. このプロジェクトをすべてのサブモジュールでクローンします
git clone https://github.com/nihui/waifu2x-ncnn-vulkan.git
cd waifu2x-ncnn-vulkan
git submodule update --init --recursive
  1. cmakeで構築します
  • MacOSのVulkan Loaderライブラリをリンクしないように、-duse_static_moltenvk = on optionで合格できます
mkdir build
cd build
cmake ../src
cmake --build . -j 4

waifu2x-caffe-cuiとの速度比較

環境

  • Windows 10 1809
  • AMD R7-1700
  • NVIDIA GTX-1070
  • Nvidiaドライバー419.67
  • CUDA 10.1.105
  • cudnn 10.1 
 Measure-Command { waifu2x-ncnn-vulkan.exe - i input.png - o output.png - n 2 - s 2 - t [ block size ] - m [ model dir ] }
 Measure-Command { waifu2x-caffe-cui.exe - t 0 -- gpu 0 - b 1 - c [ block size ] - p cudnn -- model_dir [ model dir ] - s 2 - n 2 - m noise_scale - i input.png - o output.png }

cunet

画像サイズターゲットサイズブロックサイズ合計時間GPUメモリ（MB）
waifu2x-ncnn-vulkan 200x200 400x400 400/200/100 0.86/0.86/0.82 638/638/197
waifu2x-caffe-cui 200x200 400x400 400/200/100 2.54/2.39/2.36 3017/936/843
waifu2x-ncnn-vulkan 400x400 800x800 400/200/100 1.17/1.04/1.02 2430/638/197
waifu2x-caffe-cui 400x400 800x800 400/200/100 2.91/2.43/2.7 3202/1389/1178
waifu2x-ncnn-vulkan 1000x1000 2000x2000 400/200/100 2.35/2.26/2.46 2430/638/197
waifu2x-caffe-cui 1000x1000 2000x2000 400/200/100 4.04/3.79/4.35 3258/1582/1175
waifu2x-ncnn-vulkan 2000x2000 4000x4000 400/200/100 6.46/6.59/7.49 2430/686/213
waifu2x-caffe-cui 2000x2000 4000x4000 400/200/100 7.01/7.54/10.11 3258/1499/1200
waifu2x-ncnn-vulkan 4000x4000 8000x8000 400/200/100 22.78/23.78/27.61 2448/654/213
waifu2x-caffe-cui 4000x4000 8000x8000 400/200/100 18.45/21.85/31.82 3325/1652/1236

upconv_7_anime_style_art_rgb

画像サイズターゲットサイズブロックサイズ合計時間GPUメモリ（MB）
waifu2x-ncnn-vulkan 200x200 400x400 400/200/100 0.74/0.75/0.72 482/482/142
waifu2x-caffe-cui 200x200 400x400 400/200/100 2.04/1.99/1.99 995/546/459
waifu2x-ncnn-vulkan 400x400 800x800 400/200/100 0.95/0.83/0.81 1762/482/142
waifu2x-caffe-cui 400x400 800x800 400/200/100 2.08/2.12/2.11 995/546/459
waifu2x-ncnn-vulkan 1000x1000 2000x2000 400/200/100 1.52/1.41/1.44 1778/482/142
waifu2x-caffe-cui 1000x1000 2000x2000 400/200/100 2.72/2.60/2.68 1015/570/459
waifu2x-ncnn-vulkan 2000x2000 4000x4000 400/200/100 3.45/3.42/3.63 1778/482/142
waifu2x-caffe-cui 2000x2000 4000x4000 400/200/100 3.90/4.01/4.35 1015/521/462
waifu2x-ncnn-vulkan 4000x4000 8000x8000 400/200/100 11.16/11.29/12.07 1796/498/158
waifu2x-caffe-cui 4000x4000 8000x8000 400/200/100 9.24/9.81/11.16 995/546/436

サンプル画像

元の画像

起源

ImageMagickを使用した高級2x 

convert origin.jpg -resize 200% output.png

ブラウザ

ImageMagick Lanczo4フィルターを備えたアップスケール2x 

convert origin.jpg -filter Lanczos -resize 200% output.png

ブラウザ

waifu2xノイズ= 2スケール= 2のアップスケール2x 

waifu2x-ncnn-vulkan.exe -i origin.jpg -o output.png -n 2 -s 2

waifu2x

オリジナルのwaifu2xプロジェクト

  • https://github.com/nagadomi/waifu2x
  • https://github.com/lltcggie/waifu2x-caffe

使用されるその他のオープンソースコード

  • https://github.com/tencent/ncnnすべてのプラットフォームでの高速ニューラルネットワーク推論
  • https://github.com/webmproject/libwebpすべてのプラットフォームでWebp画像をエンコードおよびデコードする
  • https://github.com/nothings/stb linux/macosの画像のデコードとエンコードのための
  • https://github.com/tronkko/direant Windows上のディレクトリにファイルをリストしてください
拡大する
追加情報
関連アプリ
おすすめ
関連情報 すべて