できること
いつも面倒な前処理を解決！！！
背景
Stable Diffusion等の画像生成AIをファインチューニングしようとした際、なるべく綺麗な画像（背景情報などが削除された画像）を使いたい。しかし、背景を削除するのは一苦労。さらに、車などの対象物のサイズもバラバラだと学習もうまくいかなかったりする。そこで、背景を削除すると同時に、対象物のサイズも揃えた画像を出力できる本スクリプトを開発。さらに、UIも開発することで誰でも簡単に画像前処理ができるようにしました。
実行方法
githubにソースコードはまとめていますので、gitに慣れている方は以降の記事をスキップして早速使ってください！以降の記事では、cloneしてUIを起動するまでを説明します。
git clone
git clone https://github.com/Jake110ii/img2removeback.git
create vertual env(任意)
pipenv shell
pipenvの参考: https://qiita.com/y-tsutsu/items/54c10e0b2c6b565c887a
module install
pip install -r requirements.txt
必要なpythonライブラリをまとめてインストールします。
run script
python main.py
これでUIの起動は完了です。http://127.0.0.1:7800/ にアクセスするとUIが起動されているはずです。どうぞ楽しい前処理Lifeをお送りください。