デザイナー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関連の情報を発信しています。