Folder Split

Posted on | In ,

폴더 나누기(Folder Split)
전체 데이터셋 폴더 -> Train, Validation, Test 폴더로 나누기

폴더 예시

변경 전

1
2
3
4
5
6
7
8
9
input/
    class1/
        img1.jpg
        img2.jpg
        ...
    class2/
        imgWhatever.jpg
        ...
    ...

변경 후

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
output/
    train/
        class1/
            img1.jpg
            ...
        class2/
            imga.jpg
            ...
    val/
        class1/
            img2.jpg
            ...
        class2/
            imgb.jpg
            ...
    test/
        class1/
            img3.jpg
            ...
        class2/
            imgc.jpg
            ...

split-folders 설치

1
pip install split-folders

사용법

비율로 나누기

  • train/val/test 나누기
    ratio 파라미터에 원하는 (train, validation, test) 비율 입력 ex) (0.6, 0.2, 0.2)
1
2
3
import splitfolders

splitfolders.ratio("input", output="output", seed=12, ratio=(.6, .2, .2))
  • train/val 나누기
    train, validation만 나누고 싶으면 두 개의 비율 입력 ex) (0.8, 0.2)
1
2
3
import splitfolders

splitfolders.ratio("input", output="output", seed=12, ratio=(.8, .2))

고정 값으로 나누기

1
splitfolders.fixed("input", output="output", seed=12, fixed=(100, 100))

https://pypi.org/project/split-folders/

0%