ailia Tech BLOG
ClothingDetection : 服装を検出する機械学習モデル
ailia SDKで使用できる機械学習モデルである「ClothingDetection」のご紹介です。エッジ向け推論フレームワークであるailia SDKとailia MODELSに公開されている機械学習モデルを使用することで、簡単にAIの機能をアプリケーションに実装することができます。
ClothingDetectionの概要
Clothing DetectionはYOLOv3を使用した服装の認識モデルです。入力された画像から、トップスの位置やボトムスの位置を検出することができます。
出典：https://github.com/simaiden/Clothing-Detection/blob/master/tests/0000003.jpg
simaiden/Clothing-DetectionAll weights and config files are in…github.com
検出可能なカテゴリ
ClothingDetectionではデータセットとしてModanetとDeepFashionV2を使用しており、入力された画像から下記のカテゴリのバウンディングボックスを計算可能です。
DATASETS_CATEGORY = {
‘modanet’: [
“bag”, “belt”, “boots”, “footwear”, “outer”, “dress”, “sunglasses”,
“pants”, “top”, “shorts”, “skirt”, “headwear”, “scarf/tie”
],
‘df2’: [
“short sleeve top”, “long sleeve top”, “short sleeve outwear”, “long sleeve outwear”,
“vest”, “sling”, “shorts”, “trousers”, “skirt”, “short sleeve dress”,
“long sleeve dress”, “vest dress”, “sling dress”
]
}
ModaNetはeBdayの提供するファッションのセグメンテーションのためのデータセットです。13のファッションカテゴリーを含んでいます。
出典：https://github.com/eBay/modanet
DeepFashionV2はファッション検出のための大規模なデータセットです。491Kの画像と13のポピュラーなカテゴリを含んでおり、合計で801Kのファッションアイテムの画像を含んでいます。
出典：https://github.com/switchablenorms/DeepFashion2
ClothingDetectionの使用方法
ailia SDKでClothing Detectionを使用するには下記のコマンドを使用します。WEBカメラから服装の認識が可能です。
python3 clothing-detection.py -v 0
アイリア株式会社はAIを実用化する会社として、クロスプラットフォームでGPUを使用した高速な推論を行うことができるailia SDKを開発しています。アイリア株式会社ではコンサルティングからモデル作成、SDKの提供、AIを利用したアプリ・システム開発、サポートまで、 AIに関するトータルソリューションを提供していますのでお気軽にお問い合わせください。