Segment Anything圖像分割模型

Segment Anything Model（SAM）從輸入提示中產(chǎn)生高質(zhì)量的物體遮罩，它可以用來(lái)為圖像中的所有物體產(chǎn)生遮罩。它已經(jīng)在一個(gè)由 1100 萬(wàn)張圖像和 11 億個(gè)遮罩組成的數(shù)據(jù)集上進(jìn)行了訓(xùn)練，并在各種分割任務(wù)中具有強(qiáng)大的性能。

安裝

需要 Python >= 3.8，以及 PyTorch >= 1.7 和 TorchVision >= 0.8。請(qǐng)按照這里的說(shuō)明來(lái)安裝PyTorch和TorchVision這兩個(gè)依賴項(xiàng)。我們強(qiáng)烈建議同時(shí)安裝支持CUDA的PyTorch和TorchVision。

安裝 Segment Anything:

pip install git+https://github.com/facebookresearch/segment-anything.git

或者在本地克隆倉(cāng)庫(kù)并安裝

git clone [email protected]:facebookresearch/segment-anything.git
cd segment-anything; pip install -e .

以下是遮罩后處理、以 COCO 格式保存遮罩、實(shí)例 notebooks 和以 ONNX 格式導(dǎo)出模型所需的可選依賴。運(yùn)行實(shí)例 notebooks 也需要 jupyter。

pip install opencv-python pycocotools matplotlib onnxruntime onnx