yolov8 전이 학습 시키기

 

 

 

 

옛날에 yolov3 시절에 마스크 검출 머신을 darknet으로 전이학습 해봤던 기억이 있다.

그 때 darknet 사용법이 좀 복잡했던 거로 기억하는데 yolov8은 엄청 쉬워졌다.

 

 

1. 파이썬이 있다고 가정하고

 

2. 레포 클론하고

git clone https://github.com/ultralytics/ultralytics

 

3. 레포 들어가서 help

yolo help

치면 다음과 학습용 명령어를 볼 수 있다.

yolo train data=xxx.yaml model=yolov8n.pt epochs=10 lr0=0.01

 

4. 여기서 data=xxx.yaml 의 xxx.yaml은

train: D:/datasets/smoking.v3i.yolov8/train/images val: D:/datasets/smoking.v3i.yolov8/valid/images test: D:/datasets/smoking.v3i.yolov8/test/images nc: 1 names: ['smoking']

이렇게 작성해 주면 된다.

 

train, val, test는 이미지 위치 적어주고

nc는 class num

names에는 class name으로 상황에 맞게

커스터마이즈 해줘야 한다.

 

yolo help 치면 predict 관련 명령어도 있어

코드 없이 간단하게 테스트 해볼 수 있다.