체크포인트 (백본별 모델·ONNX)
체크포인트 (백본별 모델·ONNX)
SAM은 백본 크기에 따라 세 가지 모델을 제공해요. sam_model_registry["<model_type>"]에 모델 타입을 지정해 로드하면 돼요.
모델 체크포인트
default또는vit_h— ViT-H SAM 모델: https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pthvit_l— ViT-L SAM 모델: https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pthvit_b— ViT-B SAM 모델: https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth
인스턴스화:
from segment_anything import sam_model_registry
sam = sam_model_registry["<model_type>"](checkpoint="<path/to/checkpoint>")
ONNX 내보내기
SAM의 경량 마스크 디코더는 ONNX로 내보낼 수 있어요. ONNX 런타임을 지원하는 환경(브라우저 포함)에서 실행 가능해요.
python scripts/export_onnx_model.py --checkpoint <path/to/checkpoint> --model-type <model_type> --output <path/to/output>
in 브라우저 데모는 demo/ 폴더의 React 앱에서 ONNX 모델로 마스크 예측을 실행해요. ONNX 내보내기에는 최신 안정 PyTorch 버전을 권장해요.