衣服の種類を判定する学習モデルについて


EIGHT
衣服の種類を判定する学習モデルについて

衣服の図面画像を読み込ませるとその種類を返すシステムを作成した。

学習モデルについて

画像から輪郭だけを抽出した輪郭画像と、オリジナル画像の2つを入力データにすることにした。

例えばTシャツならTの字になっているし、スカートなら台形になっている。 このように、衣服の種類はその輪郭である程度判定できるであろうと考えた結果、輪郭写真も入力データとして使うことを決めた。

この2枚の画像によって衣服の種類を判定させようというのが今回作成したモデルである。 実装はpytorchで行った。

各画像をCNNに入力し、特徴量を1000個抽出する。 合計2000個の特徴量をNNにより更に学習させて、最終的に9個のパラメータにする。

事前に0番目をスカート、1番目をシャツ等と割り振りさせておいて、9個の数値のうち、最も数値が高い番号を予測結果とする。 最後に予測番号を割り振った名称に置き換えて結果として返す。 CNNの部分に関してはResNet50という技術を用いるなどの工夫をした。

前処理について

まず、図面画像を.pngや.jpegからコンピュータが計算しやすいtensor形式に変換して、画像の大きさを揃える必要がある。 その上で、学習モデルに入れるために輪郭だけ抽出するなどの処理も作成した。


株式会社ファントムへのお問い合わせ

群馬県でPythonを使ったAIやソフトウェアを開発している株式会社ファントムが運営しています。




    Related Articles

    Python

    Pythonのenumerate関数でfor文のインデックスを取得

    Pythonのenumerate関数でfor文のインデックスを取得 enumerate関数を使うとPythonでforを書くときに処理毎にカウントアップしたり、値を代入したりといった処理が簡単にできるので便利です。 実際 […]

    Posted on by press
    Django

    DjangoのFormに初期値を設定する

    DjangoのFormに初期値を設定する Djangoで生成したFormをhiddenにしたまま、あらかじめ設定しておいた初期値が入ったvalueをPOSTする方法です。 forms.pyでinitial=’ […]

    Posted on by press

    最新情報をお届けします!

    メーリングリストに登録するとファントムの最新情報をお届けします

    お客様のメールアドレスを共有することはありません