Help us understand the problem. What is going on with this article?

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

はじめに

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

やりたいこと

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

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

インストール方法

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
kotai2003
Chief Technology Officer at Tomomi Research Inc. Engineering First!
https://www.tomomi-research.com
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away