sam-cpp-python 0.0.1

Python binding for sam.cpp

Python Bindings for sam.cpp

Installation

From pypi

pip install sam_cpp_python

From source

git clone https://github.com/neka-nat/sam-cpp-python.git
cd sam-cpp-python
pip install .

Example

import cv2
import numpy as np

import sam_cpp

image = cv2.imread("img.jpg")
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

predictor = sam_cpp.create_sam_predictor()
predictor.set_image(image)
masks = predictor.predict(np.array([10, 10], dtype=np.float32))

print(masks)

Result