YOLOv8 - segmentation

Image Segmentation

Image Segmentation은 이미지 픽셀 단위로 데이터를 추출하는 방법이다.
이미지에서 개체의 위치와 모양, 각 픽셀이 속하는 특정 객체 등을 알아내기 위해 이미지를 분할해서 각 픽셀에 레이블을 부여하는 것이다.

ultralytics github에서 YOLOv8 pretrained segment model에 대한 소개를 볼 수 있다.
여러 모델 중 size, speed 등을 확인하고 적절한 것을 선택해서 사용하면 된다.

첫번째 YOLOv8n-seg를 사용하여 /home/sample.jpg의 segmentation 결과를 확인하고 싶다면 다음과 같이 실행하면 된다.

yolo task=segment mode=predict model=yolov8n-seg.pt source=/home/sample.jpg

아래와 같은 오류 메시지가 나온다면

Command 'yolo' not found

다음 명령을 입력한 후 다시 실행한다.

pip install ultralytics

정상적으로 실행되면 아래와 같이 segmentation 정보가 담긴 이미지가 저장되었다고 출력된다.

Speed: 4.4ms preprocess, 118.1ms inference, 1.8ms postprocess per image at shape (1, 3, 640, 640)
Results saved to /home/runs/detect/predict

저장된 이미지는 출력된 경로(/home/runs/detect/predict)에서 확인할 수 있다.