実現したいこと
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等で検索した
- ソースコードを自分なりに変更した
- 知人に聞いた
- その他
上記の詳細・結果
同様の質問は見つけられなかった
補足
特になし
segment-anything パッケージをインストールする必要があります。(依存関係にある PyTorch パッケージ等も同様にインストールする必要があります)
segment-anything · PyPI
https://pypi.org/project/segment-anything
まずは、melianさんの指摘どおりだと思いますが、cudaも必要のようですし、この記事だけ見て動かせるようにするのはかなり厳しいのではないかと思います。
書籍などで、環境構築から機械学習を動かしてみるようなものを一通りやってから取り組まれるのがいいのじゃないかと思います。
質問のリンク先のページでも紹介されている、segment-anythingの本家のサイトには、事前準備を含めたインストール方法が書いてありますが、それは読みましたか?
https://github.com/facebookresearch/segment-anything#installation
コメントありがとうございます!
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)
ということは、pythonもしくは仮想環境は複数インストール/設定されていて、環境が整っていない方のpythonで実施しようとしているということなのではないかと思うのですが、pythonは複数インストールされていますか?
![guest](/img/icon/icnUserSample.jpg)
あなたの回答
tips
プレビュー