Google Colab で はじめる ERNIE-ViLG

「ERNIE-ViLG」が二次元キャラの画像生成が得意らしいということで試してみました。

1. ERNIE-ViLG

「ERNIE-ViLG」は、中国の大手インターネットおよび AIサービス企業のBaiduが開発した、テキストから画像を生成する画像生成AIです。Baiduが開発した深層学習フレームワーク「PaddlePaddle」とモデルハブ「PaddleHub」経由で利用できます。

2. Colabでの実行

Colabでの実行手順は、次のとおりです。GPUなくても動きます。

(1) パッケージのインストール。

!pip install paddlepaddle paddlehub
import paddlehub as hub
module = hub.Module(name="ernie_vilg")

(2) テキストから画像の生成。
今回は、猫耳メイド(貓耳女僕)で画像生成してみました。

module.generate_image(
    text_prompts=["貓耳女僕"], 
    style="卡通", 
    output_dir='./ernie_vilg_out/')

パラメータは、次のとおりです。

指定したフォルダ（今回はernie_vilg_out）に画像が出力されています。

3. 関連