肌、服、髪セグメンテーションお試し🦔

肌、服、髪のセマンティックセグメンテーションモデルを訓練していました。

データセットは自前で、モデルはDeepLabV3+、PAN(Pyramid Attention Networ)、N-Net++です。

pth形式とonnx形式で用意しています。

300枚ほどアノテーションしたので、ちょっと精度味見、、、🤔
やっぱり、Object Detectionに比べてアノテーション大変だなー👻 pic.twitter.com/VbKzRjFnmw

— 高橋 かずひと@孫請級プログラマー🦔 (@KzhtTkhs) 2021年9月7日

肌や服の検出モデルなので若干取り扱い注意な気がしますが、以下にソースコード、モデルを整理してコミットしています。

なお、今回は、データセットの収集は以下のリポジトリを活用し、

アノテーションは以下のリポジトリを活用しています。