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

試したこと・調べたこと

上記の詳細・結果

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

補足

特になし

コメントを投稿

0 コメント