【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



관련 자료


  • [Split Folders] htps : // 기주 b. 코 m / j 펑 r / sp ぃ t 푸 l로 rs
  • 좋은 웹페이지 즐겨찾기