Segment Anything 使用方法

安裝Segment Anything: 克隆GitHub倉庫並使用pip安裝包：git clone [email protected]:facebookresearch/segment-anything.git; cd segment-anything; pip install -e .

下載模型檢查點: 從Segment Anything GitHub倉庫下載預訓練模型檢查點。

導入所需模塊: 導入必要的模塊：from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor

加載模型: 使用檢查點加載SAM模型：sam = sam_model_registry['<model_type>'](checkpoint='<path/to/checkpoint>')

準備輸入圖像: 加載並預處理您的輸入圖像：image = cv2.imread('<image_path>'); image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

生成掩碼: 使用SamAutomaticMaskGenerator生成掩碼：mask_generator = SamAutomaticMaskGenerator(sam); masks = mask_generator.generate(image)

提供提示（可選）: 為了更精確的分割，提供點或框等提示來引導模型：predictor = SamPredictor(sam); predictor.set_image(image); masks, _, _ = predictor.predict(<input_point>, <input_label>)

可視化結果: 使用matplotlib或其他可視化工具在原始圖像上顯示生成的掩碼