SAM 라이브러리를 활용한 SAM 활용 실습

앞에서 탐구한 이론적 지식을 바탕으로 실제 실습에 착수할 준비가 되었습니다. 우리는 이미 SAM 모델이 어떻게 대규모 데이터 세트, 특히 SA-1B를 활용하여 이미지 분할 분야에서 혁신을 이루고 있는지를 배웠습니다. 이제는 이러한 개념들을 실제로 적용하고, 모델의 성능을 직접 평가해볼 시간입니다.

이 실습의 주된 목적은 SAM 모델을 사용하여 다양한 이미지에 대한 분할 작업을 수행하고 SAM의 프롬프트 인코더 부분을 동작해보는 것입니다.

먼저 다음처럼 SAM 라이브러리와 필요한 파이썬 패키지를 설치합니다. 그리고 SAM 모델의 사전 훈련된 가중치도 다운로드합니다.

!pip install -q 'git+https://github.com/facebookresearch/segment-anything.git' # ①
!pip install -q jupyter_bbox_widget dataclasses-json supervision               # ②
!wget -q https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth  # ③