画像内の顕著なオブジェクトをセグメントしてくれるモデル。
手順
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のモデルでこの性能。すごい。
🐣
お仕事のご相談こちらまで
rockyshikoku@gmail.com
Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。