本地测试Segment Anything

一、下载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就可以选择不是按照图片中选择；
点到最后的模板就可以看其处理生成掩膜的结果了；