rembgを利用して、生成した画像の背景を削除してキャラクターの画像のみを切り出す手順を紹介します。
Stable Diffusionで生成した画像で、キャラクターのみを利用したい場合、背景を削除してキャラクター部分のみの画像を抜き出したいです。
この記事では、rembgを利用して生成した画像から背景を削除する手順を紹介します。
rembgをインストールします。インストール手順はこちらの記事を参照してください。
画像を生成します。
今回は下図の画像を生成しました。
生成した画像のプレビューエリアの下部のボタンの[Send to extras]ボタンをクリックします。
[Extras]タブが選択され、下図の画面が表示されます。生成された画像が設定された状態になります。
ページを下にスクロールします。[Remove background]のドロップダウンリストボックスをクリックします。
ドロップダウンリストが表示されますので、背景を削除するアルゴリズムを選択します。
[u2net]の項目をクリックして選択します。
[Generate]ボタンをクリックします。処理が実行され、結果が右側のエリアに表示されます。背景が削除され、キャラクターのみの画像が表示できました。
キャラクターを抽出した画像は、透過PNG画像で取得できます。
背景画像を削除してキャラクターを抽出できました。
[Remove background]のアルゴリズムを変えて結果がどう変わるか確認します。
背景がきれいに削除できています。
後ろ髪部分に背景の削除残りが見えますが、おおむね良好です。
人間のキャラクターのみを抜き出すアルゴリズムです。手に持っているカップが含まれない結果になっています。
衣類だけが抽出される結果となりました。
きれいに背景が削除できています。
後ろ髪部分に若干背景が残っている部分がありますが、おおむね良好です。
後ろ髪部分に背景が少しだけ残っていますが、おおむね良好です。
