语义分割 SAM 部署与测试

语义分割 SAM 部署与测试_第1张图片
分割所有图片,强大的分割能力 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:

https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth

vit_b:

https://dl.fbaipublicfiles.com/segment_

你可能感兴趣的:(pyqt5,机器视觉,深度学习,pytorch,python)