デザイナーAIを育てよう

企業やブランドのロゴは、人々の心に訴えかける精緻なデザインが必要です。
AIに世の中のたくさんのロゴを学習させて、人々の心に訴えかけるデザイン性のある新しいロゴアイデアを作ってみましょう。

ロゴデータを集める

まずは綺麗なロゴの画像をたくさん集めます。
今回は架空のロゴ画像を20000枚集めました。

AIに学習させる

集めたロゴ画像をAIに学習させます。

AIに画像生成してもらう

ロゴの特徴をうまく学んだAIは全く新しいロゴを生成できます。

今回は、データセットのばらつきのせいか、GPUの課金が足りなかったせいか、うまく学習できませんでした。。。

具体的なプログラミング

今回はStyleGANという画像生成モデルを使います。

StyleGANのインストール

git clone https://github.com/autonomousvision/stylegan-xl.git
cd stylegan-xl
pip install dill
pip install ftfy
pip install timm==0.5.4
pip install Ninja
pip install setuptools==59.5.0
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

データセットの準備

画像をたくさん集めてディレクトリに入れます。
データセット作成スクリプトを実行します。

python dataset_tool.py --source=path_to_images_dir --dest=images-256x256.zip --resolution=256x256

学習

このスクリプトで学習を始められます。

python train.py --outdir=checkpoints --cfg=stylegan3-t --data=images-256x256.zip --gpus=1 --batch=64 --mirror=1 --snap 10 --batch-gpu 8 --kimg 10000 --syn_layers 10

画像生成

学習したAIで画像を生成します。

python gen_images.py --outdir=out --trunc=0.7 --seeds=10-15 --batch-sz 1 \
  --network=network-snapshot.pkl

動画生成

python gen_video.py --output=out.mp4 --trunc=0.7 --seeds=200-231 --grid=4x2 \
  --network=network-snapshot.pkl

今回は失敗しましたが、キャラクターなども生成できます。

🐣

フリーランスエンジニアです。
もし以下のようなご要望をお持ちでしたらお気軽にご相談ください。
AIサービスを開発したい、ビジネスにAIを組み込んで効率化した
い、AIを使ったスマホアプリを開発したい、
ARを使ったアプリケーションを作りたい、スマホアプリを作りたいけどどこに相談したらいいかわからない…

いずれも中間コストを省いたリーズナブルな価格でお請けできます。

お仕事のご相談はこちらまで
rockyshikoku@gmail.com

機械学習やAR技術を使ったアプリケーションを作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium
GitHub