absoluteCLF

xceptionモデルからの転移学習による分類モデル作成の自動化ツール

強力なData augumentation機能

準備

images/train および images/valid に
カテゴリ数と同数のフォルダを作成し、画像を配置する

absoluteCLF
|
|-images
|    |-train
|    |   |-classA
|    |   |-classB
|    |
|    |-valid
|    |   |-classA
|    |   |-classB
|    |
|    |-test
|        |-classA
|        |-classB
|-src

使い方

• トレーニングの開始
  python train.py -p -t

• images/testフォルダ内の画像に対して推論
  python test.py

• 指定フォルダ内の画像に対して推論
  python predict.py -f FOLDER -o OUTPUT.csv

CONFIG

config.pyへ設定を記述する