RMBG-1.4

生成AI関連

RMBG-1.4について

RMBG-1.4とは、BRIA AIによって開発された最先端の背景除去モデルです。このモデルは、一般的なストック画像、eコマース、ゲーミング、広告コンテンツなど、様々なカテゴリーと画像タイプの前景と背景を効果的に分離するために設計されています。商用利用は、商業契約が必要です。非商用利用の場合は、Creative Commonsライセンスの下でリリースされます。このモデルは、Pythonで実装されており、PyTorchを用いて学習されています。このモデルの品質は、RemBGやremove.bgなどの他の背景除去ツールよりも高いと言われています。このモデルを利用するには、Hugging Faceのサイトからダウンロードするか、デモページで試すことができます。

RMBG-1.4ツールのインストール

インストールフォルダを作成します。

管理者モードでcmdコマンドを実行します。

作成したインストールフォルダにcdして移動します。

PyTorchをインストールします。

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

そして、デモを丸ごと取得します。

git clone https://huggingface.co/spaces/briaai/BRIA-RMBG-1.4

クローンできたら、リポジトリルートに移動しておきます。

cd BRIA-RMBG-1.4

フォルダに出来上がっている以下をインストールします。

requirements.txt

pip install -r requirements.txt

補足:自分の環境では以下のエラーがでました。

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
tensorflow-intel 2.13.0 requires typing-extensions<4.6.0,>=3.6.6, but you have typing-extensions 4.9.0 which is incompatible.

バージョンのアンマッチが起きてるようですが、動かして見ると問題無く動きました。上手く動かない場合はエラーに合わせたバージョンダウンをして下さい。

RMBG-1.4ツールの動作確認

毎回コマンドプロンプトから実行するのは面倒なのでbatchファイルを作成します。

  1. メモ帳などのテキストエディタを開きます。
  2. 下記のコマンドをテキストファイルに入力します。(正常に動作しない可能性があるので2バイト文字を含むフォルダは使わないで下さい。)
@echo off
C:
cd \BRIA-RMBG\BRIA-RMBG-1.4
python app.py
pause

C:の部分やフォルダパスの部分は自分の環境に合わせて適宜修正して利用して下さい。

作成したbatchファイルを起動してしばらく待つとローカルのアドレスが表示されます。そのアドレスにブラウザでアクセスすれば利用可能になります。

「http://127.0.0.1:7860」にアクセスしたら、あとは普通に利用できます。

コメント

タイトルとURLをコピーしました