商用利用可能な画像生成AI Python
商用利用可能な画像生成AIにはいくつかの選択肢があります。これらのモデルは、テキストから画像を生成するために使用できます。代表的なものには、OpenAIのDALL-EやStability AIのStable Diffusionなどがあります。
ここでは、Stable DiffusionをPythonで使用する方法について説明します。Stable Diffusionはオープンソースであり、商用利用も可能です。
まず、必要なパッケージをインストールします。これには、diffusersライブラリを使用します。
bash
pip install diffusers transformers
次に、Stable Diffusionモデルを使用して画像を生成するPythonコードを示します。
python
import torch
from diffusers import StableDiffusionPipeline
# Stable Diffusionのパイプラインをロード
pipeline = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
pipeline = pipeline.to("cuda")
# テキストプロンプトから画像を生成
prompt = "A fantasy landscape with mountains and a river"
image = pipeline(prompt).images[0]
# 生成した画像を保存
image.save("generated_image.png")
このコードでは、Stable Diffusionモデルをロードし、指定されたテキストプロンプトに基づいて画像を生成しています。生成された画像は、generated_image.pngというファイルに保存されます。
注意点
-
GPUの使用: モデルは非常に大きく、画像生成には高い計算リソースが必要です。上記のコードではCUDA対応のGPUを使用する設定になっています。GPUがない場合は
"cuda"の部分を"cpu"に変更してくださいが、生成時間が大幅に長くなります。 -
ライセンスと利用規約: 商用利用を行う場合は、モデルのライセンスと利用規約を確認し、遵守する必要があります。
これで、商用利用可能な画像生成AIをPythonで使用する基本的な方法が分かります。次に進むには、具体的なニーズに応じてモデルのチューニングや追加の設定を行うことを検討してください。