반응형
자세한코드는 깃헙참고
init으로 나의 현재 디렉토리의 경로를 준다.
디렉토리 생성
train파일안에 각class의 파일이 생긴다.
만들어둔 각 file안으로 각이미지를 train, valid, test를 7:2:1의 비율로해서 옮겨준다.
run함수
mk_dir함수와 move_img함수를 실행시켜준다.
checking_dirs함수
잘옮겨졌는지 확인해주는함수
여기까지 실행시키면 각file별로 이미지들이 잘 옮겨진다.
728x90
'computer vision > dna_study' 카테고리의 다른 글
pytorch 데이터셋 클래스 (0) | 2023.03.17 |
---|---|
python 이미지생성 (0) | 2023.03.16 |
python train test vaild 폴더 생성 (0) | 2023.03.15 |
kaggle Tutorial Intel Image Classfication(캐글) (1) | 2023.03.15 |
CV(Computer Vision) 전이학습(Transfer Learning) (0) | 2023.03.03 |
댓글