インスタンスセグメンテーション YOLACT

同じ種類の物体でも個体ごとに分けて認識するインスタンスセグメンテーション。
高速なYOLACT。

人物が全て別々のインスタンスとして認識されている。

使い方

インストール。

git clone https://github.com/dbolya/yolact.git
cd yolact
pip install cython

リポジトリから事前トレーニング済み重みをダウンロードして実行。

python eval.py --trained_model=yolact_im700_54_800000.pth --score_threshold=0.15 --top_k=15 --image=your_image.png:output_image.png

マスクした画像が保存される。

🐣

---

フリーランスエンジニアです。
お仕事のご相談こちらまで
[e-mail]

機械学習、ARアプリ（Web/iOS）を作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium
GitHub