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や他の視覚化ツールを使用して、生成されたマスクを元の画像に表示します