segment anythingの使い方について

実現したいこと

https://qiita.com/john-rocky/items/5fc05d96c4c9f70f6e1b
に書かれていることをやってみたい

発生している問題・分からないこと

from segment_anything import sam_model_registry, SamPredictor
ModuleNotFoundError: No module named 'segment_anything'
というエラーの解決方法を知りたい

該当のソースコード

python
1from segment_anything import sam_model_registry, SamPredictor
2import torch
3
4sam_checkpoint = "sam_vit_h_4b8939.pth"
5model_type = "vit_h"
6
7device = "cuda"
8
9sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
10sam.to(device=device)
11
12predictor = SamPredictor(sam)
13

試したこと・調べたこと

teratailやGoogle等で検索した
ソースコードを自分なりに変更した
知人に聞いた
その他

上記の詳細・結果

同様の質問は見つけられなかった

補足

特になし

melian

2024/07/04 07:26

segment-anything パッケージをインストールする必要があります。(依存関係にある PyTorch パッケージ等も同様にインストールする必要があります) segment-anything · PyPI https://pypi.org/project/segment-anything

TakaiY

2024/07/04 09:42

まずは、melianさんの指摘どおりだと思いますが、cudaも必要のようですし、この記事だけ見て動かせるようにするのはかなり厳しいのではないかと思います。書籍などで、環境構築から機械学習を動かしてみるようなものを一通りやってから取り組まれるのがいいのじゃないかと思います。

TakaiY

2024/07/04 09:54

質問のリンク先のページでも紹介されている、segment-anythingの本家のサイトには、事前準備を含めたインストール方法が書いてありますが、それは読みましたか？ https://github.com/facebookresearch/segment-anything#installation

n.k_0220

2024/07/04 09:59

コメントありがとうございます! pytorch,cuda,cudnnはインストール済みです。 pip install segment-anythingでインストールはしています。出力は以下の通りです。 Requirement already satisfied: segment-anything in c:\users\appdata\local\programs\python\python39\lib\site-packages (1.0)

TakaiY

2024/07/04 10:14 編集

ということは、pythonもしくは仮想環境は複数インストール/設定されていて、環境が整っていない方のpythonで実施しようとしているということなのではないかと思うのですが、pythonは複数インストールされていますか？