1. 이미지 및 경로 세팅

1-1. 학습할 객체의 다양한 조건에 맞는 이미지를 여러 개 수집한다.

image.png

1-2. 다음과 같이 폴더를 생성하고 이미지를 …/yolo_custom/dataset/train/images 폴더에 넣는다.

(본인은 C 드라이브 안에 yolo_custom 폴더를 넣음)

/yolo_custom
    /datasets
		    /train
            /images
            /labels
        /val
            /images
            /labels

2. labelimg 라이브러리 설치 및 세팅

2-1. 데이터 라벨링을 위해 labelimg 라이브러리 설치

(데이터 라벨링이란 기계 학습 모델을 훈련시키기 위해 데이터에 적절한 레이블(정답)을 붙이는 과정이다.)

pip install labelimg

2-2. labelimg 실행

labelimg

2-3. Open Dir 버튼을 클릭하여 이미지를 넣은 …/train/image 폴더 선택

스크린샷 2024-11-24 162924.png