U-2Netをつかう。

画像内の顕著なオブジェクトをセグメントしてくれるモデル。

#手順

#####1、U2-NetのGitHubリポジトリをクローン

git clone https://github.com/NathanUA/U-2-Net.git
cd U-2-Net/

#####2、事前学習済みモデルをリポジトリのリンクからダウンロードし、指定のディレクトリに配置

mkdir saved_models/u2netp/

#####3、とりあえずPythonで試してみる。
リポジトリで配布されているモデルは１７４MBと４MBの２種類あって、軽量の方をつかう場合は、model_nameをu2netpに書き換えます。

あと、PytorchでデフォルトはCudaデバイス（GPU）をつかう設定になっているので、CPUで実行する場合はtorch.deviceを切り替えます。

u2net_test.py

model_name='u2net'#u2netp

python u2net_test.py

４MBのモデルでこの性能。すごい。

🐣

---

お仕事のご相談こちらまで
[e-mail]

Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。

Twitter
Medium
ホームページ