2020 졸업프로젝트 - 기술 기록용

labelImg 라이브러리를 이용하여 데이터셋 만들기

210_yy 2021. 2. 15. 16:09

※ 주의 ※ 글쓴이는 mac os를 사용합니다

 

 

모델 훈련을 위해 나만의 데이터셋을 만들 때, 직접 사진을 다운받고 라벨링을 하고자 한다 ! 

-> labelImg 라이브러리를 사용하면 된다 

 

 

1. labelImg 라이브러리 설치

 

terminal에서 

pip install labelImg 입력

 

(너무 옛날에 다운받아서 정확히 기억은 안나지만 ㅠㅠ 이렇게 했었던것 같아서,,)

 

 

2. labelImg 실행

 

terminal 창에서 

 

labelImg 를 입력하면

 

 

 

 

 

이러한 창이 뜬다.

 

 

3. 라벨링 하기

 

Open Dir 로 사진들이 저장된 디렉토리를 선택하면

 

첫번째 사진이 뜨고 오른쪽 아래에는 파일들의 리스트가 보여진다.

 

바운딩 박스 정보 저장은 -> Change Save Dir 를 통해 경로 설정 

 

 

우리는 yolo 데이터셋을 만들것이므로 

 

메뉴바에 PascalVOC버튼을 한번 누른다.

 

 

그럼 이렇게 yolo 버튼으로 바뀜 

 

 

 

 

create\nRectBox를 누르고 라벨링 시작 (모든 차를 박스)

 

 

 

라벨링을 정확하게 하는것도 모델 정확도에 영향을 끼칠 것 같다고 우리 팀원들은 생각했다.. (처음 할 때 박스를 정확하게 그립시다 ㅠㅠ) 

 

이렇게 노가다로 모든 사진을 라벨링 해주면 된다 ㅎㅎ.. 

반응형