根据官方的readme即可解决大多数问题,难得部分是本地资源的查找
创建虚拟环境python3.7以上
安装torch并git源码https://github.com/luca-medeiros/lang-segment-anything.git
到有pyproject.toml的目录下,pip install -e .
手动git并复制segment-anythin;groudingdino到主文件下
建立demo.py并补全缺失的包,内容如下
from lang_sam import LangSAM
model = LangSAM("vit_b","./sam_vit_b_01ec64.pth")
具体分割效果可查找其他教程
以下为需要修改的本地查找部分
在lang_sam文件夹下的lang_sam.py
找到load_model_hf,将涉及cache的目录都注释或删除,替换为本地文件
修改LangSAM类,或者实例化时sam_type进行本地模型及地址的索引
提前下载bert-base-uncased,groudingdino有所使用,若无法指定下载哪几个,可以全部下载了先。