[YOLO] 소량의 데이터셋 경로를 포함하는 text 파일 생성

코드 주요 사용법

 

프로젝트

Sonk_Program

 

사용되는 파일 경로

utils/sorting.py

utils/filPath.py

utils/fileIO.py

 

 

mode_number 0, 1, 2 를 사용하여 train, validation, test dataset을 구분하여 text 파일을 생성한다.

 

과정:

1) mode_number를 사용자 변수로 넣고 sorting.py를 실행한다. 

2) mode_number로 train / val / test 를 구분한다.

3) count_data_per_class(text 라벨 폴더 경로, img 파일이름이 저장된 text 경로, 파일 확장자, (클래스 이름들) ) 

--> class id or name : num of object 가 dictionary 형태로 리턴.

4) num of object가 오름차순으로 정렬된 file prefix, 각 class's object의 시작 위치를 리스트로 리턴

5) file prefix 리스트, text 라벨 폴더 경로, num of obj가 있는 dictionary, divisor, 파일 확장자를 매개변수로 divide_aligned_list 함수 실행함 --> divisor 만큼 나눠진 새로운 prefix 리스트를 리턴

6) 새로운 prefix 리스트 저장할 text 경로, 이미지 데이터셋 폴더 경로, 새로운 prefix 리스트, 파일확장자 매개변수로

write_image_path_from_list() 실행 --> Reduced dataset의 이미지 경로들을 지정한 경로에 저장함.