코드를 실행하면 다음처럼 출력됩니다.

▲ 그림 5-51 출력 결과: SAM 모델의 이미지 분할 영역 개별 시각화

이렇게 간단하게 SAM을 이용하여 만능 분할 모델을 구현해보았습니다. SAM의 큰 특징 중 하나는 프롬프트를 입력받아 원하는 부분에 대한 분할을 진행할 수 있습니다. SamPredictor 클래스는 모델에 프롬프트를 표시하기 위한 쉬운 인터페이스를 제공합니다. 이를 통해 사용자는 먼저 필요한 이미지 임베딩을 계산하는 set_image 메서드를 사용하여 이미지를 설정할 수 있습니다. 그런 다음 예측 메서드를 통해 프롬프트를 제공하여 해당 프롬프트에서 마스크를 효율적으로 예측할 수 있습니다. 모델은 점 및 상자 프롬프트를 모두 입력으로 받을 수 있습니다.