거북이의 공부/ML

데이터 학습_Labeling Tool(labelImg)

두부막냉누나 2021. 11. 3. 11:59

내가 원하는 데이터를 학습 시키기 위해서는 조금의 노가다(?)가 필요합니다..ㅎ

제가 쓰고 있는 labeling tool은 바로 'LabelImg'

 

사진에서 원하는 영역을 선택한 뒤 label을 선택하면 끗..! 간단하죠? 

LabelImg 사용법 (예시는 제 강아지 두부ㅎㅎ)

 

아주아주 간단한 labelimg 사용법 알려드릴게요!

 

1. labelImg 설치

1) pip install labelImg 만 해주면 끗!

$ pip install labelImg   

2) https://tzutalin.github.io/labelImg/

두번째 방법은 해당 링크 들어가서 본인 컴퓨터에 맞게 설치해주시면 됩니다.

 

2. labelImg 실행 

$ labelImg     

입력만 하면 자동으로 labelImg가 실행됩니다.

3. Open Dir

레이블링 하고자 하는 데이터 셋 선택과 좌표값이 저장될 위치 선택!

 

4. 학습할 format으로 변경해주기

저는 yolo로 학습할 거라 yolo 선택했습니당 

5. 원하는 영역 드래그

저는 두부의 얼굴을 학습시키고 싶어서 얼굴 체크! 한 다음 두부라고 레이블 달아줬어요 ㅎㅎ

 

auto save로 setting 바꿔주시면 끗!!!!!!!

레이블 디렉토리 가시면 사진 원본 title과 똑같은 txt파일이 생성된 것을 확인 할 수 있어요!

 

단축키 tip!

w - create RectBox

d - next Image

a - previous image

 

 

참고: 

https://github.com/tzutalin/labelImg