공부/Python

[PJT] ImageLabelingTool

래울 2024. 5. 30. 01:22

NDIVIA에서 Jetbot의 Road Following을 위해 기본으로 제공하는 데이터 수집 및 라벨링 코드가 있는데, 시간도 너무 오래걸리고

귀찮아서 개발했다.

(원래 생각만하고 할까 말까 하고 있었는데, 프로젝트 같이 했던 조원이 등 떠밀어 준 덕분에 얼떨결에 만듬)

 

PyQt + OpenCV 로 만듬, 원래는 Object Detection 학습에도 확장하여 사용할 마음으로 Type 옵션도 넣어놨는데,

프젝할 시간이 부족해서 RoadFollowing 이미지 라벨링에만 사용했다.


 

1. 사용법

Key Control
- A, D : 이미지 좌우 넘기기(이전/다음 이미지)
- W, S : 라벨링 타입 조작 (Not used), 추후 필요 시 기능 등록하여 사용 가능
  * 만약 키 입력이 되지 않는다면 GUI 라벨을 한 번 클릭 후 사용

저장 (작업 완료 후 저장 필)
- Ctrl +S 또는 상단 메뉴 바의 저장 클릭
- 저장 시, /result 폴더가 생성되며, 이미지들이 라벨링되어 저장됨

라벨링할 이미지들의 경로는 기본 값으로 ./images 로 설정됨

 

 

 

2. 실행 사진

 

라벨링 완료 된 이미지는 라벨링 값과 함께 초록색으로 바뀜

 

 

 

3. 코드

https://github.com/Hojun1123/ImageLabelingTool-For-Jetbot/tree/main

 

GitHub - Hojun1123/ImageLabelingTool-For-Jetbot: 젯봇의 Road Following을 위한 이미지 라벨링 도구

젯봇의 Road Following을 위한 이미지 라벨링 도구. Contribute to Hojun1123/ImageLabelingTool-For-Jetbot development by creating an account on GitHub.

github.com