ホームページ > 記事 > バックエンド開発 > Python を使用して写真をアニメスタイルのアバターに変換します。

Python を使用して写真をアニメスタイルのアバターに変換します。

王林転載: 2023-04-22 22:22:111969ブラウズ

最近、アバターをアニメ風に変換する必要があるという問題が発生し、そのときに最初に考えたのが、既製のホイールを探すことでした。

Python を使用して写真をアニメスタイルのアバターに変換します。

▲写真をピクセルスタイルに変換する

アバターをアニメスタイルに変換することは、元の画像情報とテクスチャの詳細を維持しながら、実際の写真を実際の写真に変換することです。アニメ/漫画スタイルの非フォトリアリスティックな画像に変換します。現在、Baidu API に加えて、Github には直接使用できるオープンソースライブラリが多数あります。

その中でも、AnimeGAN は武漢大学と湖北理工大学の研究であり、ニューラルスタイル転送と敵対的生成ネットワーク (GAN) を組み合わせて使用しており、その効果は私たちのニーズと非常に一致しています。

Python を使用して写真をアニメスタイルのアバターに変換します。

#AnimeGAN は最初 Tensorflow フレームワークを使用していましたが、情報を照会した後、プロジェクトがすでに PyTorch フレームワークをサポートしていることが判明しました。

Python を使用して写真をアニメスタイルのアバターに変換します。

アドレス: https://github.com/bryandlee/animegan2-pytorch

そして、私はたまたま以前に PyTorch に基づいた Weibo コメント感情分析を実装したことがあります。 . そのため、使用するのに負担がかからず、ライブラリをインストールする必要もありません。

Pytorch のインストール

PyTorch は、自然言語処理などのアプリケーション向けの、Torch に基づくオープンソースの Python 機械学習ライブラリです。この深層学習フレームワークは、数値モデリング、画像モデリング、テキストモデリング、音声モデリングなど、さまざまな方向に適用できます。

Python を使用して写真をアニメスタイルのアバターに変換します。

Pytorch のインストールは他のライブラリに比べて面倒なので、公式インストールしてダウンロードする場合は、実際の設定に基づいて自分に合ったインストールコマンドを入手する必要があります。

Python を使用して写真をアニメスタイルのアバターに変換します。

ダウンロード速度が遅い場合、または上記の手順でさまざまな問題が発生した場合は、次の Web サイトを試してみるとよいでしょう:

https :/ /download.pytorch.org/whl/torch_stable.html

Web サイトの表示:

Python を使用して写真をアニメスタイルのアバターに変換します。

アニメスタイルの移行

インストール後Pytorch フレームワーク。animegan2-pytorch プロジェクトをローカルに複製したり、直接ダウンロードしたりできます:

git clone https://github.com/bryandlee/animegan2-pytorch

以下に示すように、記事の最後にあるローカルディレクトリにダウンロードします:

Python を使用して写真をアニメスタイルのアバターに変換します。

# ウェイトフォルダには 4 つのウェイトが含まれています。必要なアニメーションスタイルの移行を実現するには、対応するウェイトを選択してください。画像のサンプルは、サンプルの下の入力フォルダーに保存されており、水をテストするために直接使用できます。さらに、処理された画像を保存するために、同じパスの下に新しい出力フォルダーも作成しました。

次に、コマンドラインで test.py スクリプトを実行してプロジェクトを呼び出すだけです。具体的なコマンド形式は次のとおりです:

python test.py --checkpoint [modelファイルパス] --input_dir [入力画像が配置されているディレクトリ] --output_dir [出力ディレクトリ] --device [デバイスの選択、CPU または Cuda]

実際の操作:

Python を使用して写真をアニメスタイルのアバターに変換します。