Get AI to be a designer ~ Create a brand logo ~ Image generation with StyleGAN-xl

MLBoy
3 min readMar 16, 2023

Develop designer AI

Corporate and brand logos require sophisticated designs that appeal to people’s hearts.
Let AI learn many logos in the world and create new logo ideas with design appealing to people’s hearts.

collect logo data

First, collect a lot of beautiful logo images.
This time we collected 20000 fictitious logo images.

let AI learn

Let AI learn the collected logo images.

Get AI to generate images

An AI that has successfully learned the characteristics of a logo can generate a completely new logo.

This time, it was not possible to learn well, probably due to the variation in the dataset, or because the GPU was not charged enough. . .

concrete programming

We use StyleGANxl.

https://github.com/autonomousvision/stylegan-xl

Install 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

Preparing the dataset

Collect a lot of images and put them in a directory.
Run the dataset creation script.

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

Train

You can start learning with this script.

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

Image generation

Generate images with learned AI.

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

Video generation

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

I failed this time, but you can also generate characters and so on.

🐣

I am a freelance engineer.
If you have any of the following requests, please feel free to contact us.
“I want to develop an AI service,
I want to , I want to develop a smartphone app that uses AI
, I want to create an application that uses AR, I want to create a smartphone app, but I don’t know who to consult…”

All of them can be ordered at a reasonable price without intermediate costs.

For work inquiries, please contact
rockyshikoku@gmail.com

We are creating applications using machine learning and AR technology.
We are disseminating information related to machine learning/AR.

GitHub

Twitter
Medium

--

--