【Python】多重分類問題のTraining, Validation, Testフォルダーを簡単に作る方法

#はじめに

多重分類問題(Multi Classes Classification）において、自前の画像データをTraining, Validation, Testに分ける方法を紹介します。

#やりたいこと

自分が持っている画像データが下記の図のようにinputフォルダーにあるとします。

その画像データを下記の図のようにtraining, Validation, Testフォルダーに分けたいと思います。

#インストール方法
pip install split-folders

#使い方
Pythonプログラムを作成します。split-foldersモジュールをインポートして、下記のコードを実行します。

import split_folders

# Split with a ratio.
# To only split into training and validation set, set a tuple to `ratio`, i.e, `(.8, .2)`.
split_folders.ratio('input_folder', output="output", seed=1337, ratio=(.8, .1, .1)) # default values

#関連資料

1. [Split Folders]https://github.com/jfilter/split-folders