[Python] 파이썬 라이브러리 splitfolders 사용하여 train/val/test 데이터 나누기

splitfolders 설치

pip install split-folders

 

input 폴더내에서 변하는 것이 아니라

output 폴더가 자동으로 생성되어 저장된다. 

import splitfolders 
# train:val = 8:2
splitfolders.ratio("input_folder_path", output="output_path", seed=1337, ratio=(.8, .2))
# train:val:test = 8:1:1
splitfolders.ratio("input_folder_path", output="output_path", seed=1337, ratio=(.8, .1, .1))