WindowsでAIプログラミング

Stylized Neural Painting:実装方法

はじめに

画像から油絵、カーカーペン画、水彩画、テープアートが描けます。

参考:https://jiupinjia.github.io/neuralpainter/

PC環境

Windows10 pro (Windows Insier Program)
マウスコンピューター製G-Tune E5-144
CPU:インテル(R) Core(TM) i7-10875H プロセッサー
( 8コア / 16スレッド / 2.30GHz [ 最大5.10GHz ] / 16MBキャッシュ )
メモリ:32GB メモリ [ 16GB×2 ( PC4-21300 / DDR4-2666 ) / デュアルチャネル ]
SSD (M.2):512GB NVMe SSD ( SAMSUNG PM981a / M.2 PCIe Gen3 x4 接続 )
グラフィックス:NVIDIA GeForce RTX2060 / 6GB

実装方法

Gitクローン

カレントディレクトリの移動

仮想環境

仮想環境の作成

アクティベイト

モジュールのインストール

『pip』と『setuptools』のアップデート

requirements.txtよりインストール

『Pytorch』の入れ替え

※必須ではありません。CPU版Pytorchでも動くみたいです。

『Pytorch』のアンインストール
『Pytorch(CUDA10.1対応)』のインストール

学習済みモデルのダウンロード

下記のURLからモデルをダウンロード・解凍し、
解凍フォルダごと(checkpoints_G_oilpaintbrush)
フォルダ(stylized-neural-paintin)に、コピーしてください。

軽量版もあります

実行コマンド

(りんごの画像から生成するコマンド。これ以外にもデモコマンドが実装されています。)

結果

下記の様なウィンドウが新しく立ち上がります。

『G_pred』は、リアルタイムに映像が見れます。

『output』フォルダに『apple_input.png』、『apple_final.png』、
『apple_animated.mp4』と連番の『apple_rendered_stroke_****.png』のファイルが生成されます。

『apple_input.png』

『apple_final.png』

『apple_animated.mp4』

おいらのオリジナル

この記事を書いた人
化学を勉強し博士号を取得したが、 あるとき、これからの時代はプログラミング!、と目覚め、 主にWindows環境下でPythonを用いてのAI奮闘。 最近は、フロントにVue.js、バックにFastAPIを用いて、 Pytorchを実装したWebサービスを開発中。☆('ω')☆
SNSでフォローする

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

CAPTCHA