自動で背景を削除 remove.bgを再現してみた - Qiita

少し前に話題になった「remove.bg」という背景を削除してくれるサービスを使ってみました。その精度の高さに感銘をうけ、興味を持ったため自分でも試してみました。コードはすべてこちらの記事に載せています。 remove.bgとは 自動で画像から背景を削除してくれるサービス 基本無料で使用可能、API取得は有料 髪の毛の一本まで抽出してくれる Semantic Segmentation このサービスを使ってみて、まず最初に思いついたのがセグメンテーションの利用です。なのでtorchvisionのDeepLabv3でサクッと試してみました。以下がその結果です。 ある程度上手くいっていますが毛の一本一本を抽出することはできていないようです。 セグメンテーションで背景削除ができない原因としては ・ 毛の一本一本に注目してもlossはそんなに変わらない ・そもそもデータセットのターゲット画像が大雑把