一、下载GitHub代码
官网地址: https://github.com/facebookresearch/segment-anything
git clone 或者 下载ZIP压缩包
二、下载.pth文件
官网中给出了三个训练好的参数文件
点击下载,我这里下载了最后一个358M大小的模型(这里可以使用迅雷下载但速度也不快)
三、本地测试
我这里使用的机器是window10系统并没有Navid显卡,需要安装好Anaconda+pytorch+vscode,详情可见另一篇博客;
在VScode中打开相应文件automatic_mask_generator_example.ipynb,在vscode中激活你的pytorch环境,并在vscode中安装好jupyter插件;
将下载好的.pth文件放到该文件的目录下;
找到set-up模块;
点击运行,注意这里需要你的pytorch环境安装相对应的库,这时打开Anaconda powershell Prompt(Anaconda),激活你的pytorch环境:conda activate pytorch(你的pytorch环境名称),输入conda install 库名进行安装相应库;
运行通过后,一步步点直到这个模块:
这里选择自己的图片名称,文件夹在相对应的路径下:
将其中的“sam_checkpoint改为自己要使用的.pth文件名称;
model_type选择相应的名称;
device要是安装的cuda版本的pytorch就可以选择不是按照图片中选择;
点到最后的模板就可以看其处理生成掩膜的结果了;