YOLOv5 Segmentation - roboflow

• realtime으로 이미지 객체를 식별.

• 기존의 YOLO 모델은 바운딩 박스만 구할 수 있음.

• YOLOv5는 가장 최신에 release된 버전인 YOLOv5 7.0 에서 Segmentation 모델로 제공. 기존에 다른 버전의 yolo를 쓰고 있다면 업그레이드 해야 함.

• 기존 YOLOv5 모델은 yolov5s.pt, yolov5s.yaml로 pretrained model이 제공되었다면, segmentation이 가능한 모델은 yolov5s-seg.pt 등의 이름으로 제공되고 있다.

• 해당 모델은 YOLOv5 공식 깃허브 링크에서 받을 수 있음. 성능에 따라 n, s, m, l, x로 제공됨.

• 성능이 늘어날수록 object detect 시간이 길어진다.

• 이는 webcam 등의 카메라를 사용하는 영상에서의 detcting에서 큰 단점이 될 수 있어 적절한 모델 선택이 필요하다.

예시

• 해당 모델을 다운받아 리눅스의 yolov5 폴더에 넣어주면 트레이닝 준비가 끝난다.

• 기존의 YOLOv5처럼 labelIMG 툴로 네모난 bounding box를 그리던 것과 달리, segmnetation은 다각형 올가미 도구를 사용하여 labeling 진행.

• YoLOv5 공식 깃허브에서 roboflow를 활용하여 Segmentation을 진행할 수 있는 방법을 소개.