더북(TheBook)
Processing images: 100%|██████████| 70/70 [00:00<00:00, 181.60it/s]
Processing labels: 100%|██████████| 70/70 [00:00<00:00, 2833.66it/s]
Processing images: 100%|██████████| 70/70 [00:00<00:00, 205.11it/s]
Processing labels: 100%|██████████| 70/70 [00:00<00:00, 2280.56it/s]

make_dataset 함수는 이미지 파일 경로 목록, XML 파일 경로 목록, 그리고 데이터 세트 내의 클래스 목록을 입력으로 받아, 이미지와 레이블의 데이터 세트를 생성합니다. 내부에 process_image 함수가 정의되어 있어 주어진 이미지 파일 경로로부터 이미지를 읽고, (224, 224)로 리사이즈한 후, 픽셀 값을 0과 1 사이로 정규화합니다(①).

이후에 이미지 파일 경로 목록을 순회하며 process_image 함수를 사용해 이미지 데이터 세트를 생성합니다(②). 동일하게 XML 파일 경로 목록을 순회하며 get_label_fromImage 함수를 사용하여 레이블 데이터 세트를 생성합니다(③). 생성된 이미지와 레이블 데이터 세트는 numpy 배열로 변환되며, 특히 레이블 데이터 세트는 적절한 모양으로 재구성됩니다(④). 함수의 마지막에서는 이미지 데이터 세트와 텐서플로 텐서 형식으로 변환된 레이블 데이터 세트를 반환합니다. 이 함수를 사용해 훈련용과 테스트용 이미지 및 레이블 데이터 세트를 생성합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.