Data Labeling

이현근·2023년 12월 21일

Yolo 어떻게 하는건데?

목록 보기

3/3

지도학습 객체탐지를 할 것이기 때문에 모델보단 데이터다.
학습용 데이터셋 부터 만들어보겠다.

폴더구조

- Data
  ├ images
  │  └ img_n.jpg
  ├ labels
  │  ├ classes.txt
  │  └ img_n.txt
  └ meta
     ├ data.yaml
     ├ train.txt
     ├ val.txt
     └ test.txt

학습시키고자 하는 image 를 images 에 전부 집어넣는다.

Target 이미지

예시로 사용할 이미지는 아래의 커넥터 이미지이다.

커넥터엔 구멍이 6개가 있는데 다음과 같이 분류를 해본다

Four_hole : 십자가 주변의 4개 구멍
Center_hole : 중심 구멍
Direction_hole : 십자가 중 한방향에만 있는 구멍
라벨링 하게되면 다음과같다.

라벨링 툴 Labelimg

Anaconda prompt 에서 다음을 진행한다

가상환경 실행
conda activate vision_ai

labelimg 설치
pip install labelimg

실행
label img

순조롭게 진행되었으면 다음과 같은 창이 뜬다

하지만 labelimg 는 개발이 중단되었는지 변수형식 문제가 생겨서
파일 두개를 수정해줘야... 에러가 안 뜨고 정상 작동한다.
int 입력 받는곳에 Float 들어가는 애러랑 ( float 앞에 int해주면 됨)

파일 1 경로 -
C:\Users\<user_name>\scoop\apps\anaconda3\current\App\envs\vision_ai\Lib\site-packages\labelImg\labelimg.py
# 965줄
bar.setValue(int(bar.value() + bar.singleStep() * units)) 
# 971줄
self.zoom_widget.setValue(int(value))
# 1025~1026줄
h_bar.setValue(int(new_h_bar_value))
v_bar.setValue(int(new_v_bar_value))

파일 2 경로 -
C:\Users\<user_name>\scoop\apps\anaconda3\current\App\envs\vision_ai\Lib\site-packages\libs\cnavas.py

# 526줄
p.drawRect(int(left_top.x()), int(left_top.y()), int(rect_width), int(rect_height))
# 530~531줄
p.drawLine(int(self.prev_point.x()), 0, int(self.prev_point.x()), int(self.pixmap.height()))
p.drawLine(0, int(self.prev_point.y()), int(self.pixmap.width()), int(self.prev_point.y()))