分割所有图片,强大的分割能力 SAM
本文做了简单的封装,对源程序做了简单修改,方便使用
可以在input文件夹中放入文件,在output文件夹中查看结果
其中segment-anything\notebooks\predictor_example.ipynb 拥有相关定位点的分割。
功能非常强大,部署即实现。
注意:处理时间较长 CPU预计180秒 GPU: 8秒内
GPU消耗内存:12G左右
1.下载程序代码
源码位置:https://github.com/facebookresearch/segment-anything
安装等请查看 本地部署与复现
2.下载权值文件
vit_h:
https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
vit_l: