1.什么是segment-anything？
Segment Anything Model (SAM): a new AI model from Meta AI that can "cut out" any object, in any image, with a single click.
官方网站：Segment Anything | Meta AI
官网提供了一些demo，交互式地分割展示，可以试一下
官网demo提供了3种分割方式。
Hover&Click(鼠标悬浮及点击)：
点击想要分割的内容，自动完成分割。左键正选，右键反选。
Box
鼠标圈一个方框，自动分割。
Everything
一键分割所有。
2.Github项目
GitHub网址：https://github.com/facebookresearch/segment-anything
readme很详细，可按照步骤来搞。
克隆远程仓库到本地：
git clone https://github.com/facebookresearch/segment-anything安装Segment Anything :
cd segment-anything; pip install -e .特别注意-e后面一个点
Github页面里点击下载一个或者多个模型：
模型文件放到项目的目录即可。
H,L,B分别表示huge,large,base，从大到小。根据硬件能力选择合适的模型。
创建虚拟环境及安装相应的包
3.使用自己的数据集进行分割
法1：使用官方命令
如下，建立input,output文件夹
在input中存放待分割的图片，output用作存放输出的mask。
在pycharm终端中使用下面的命令：
使用base模型
python scripts/amg.py --checkpoint sam_vit_b_01ec64.pth --model-type vit_b --input C:\GithubFile\segment-anything\input --output C:\GithubFile\segment-anything\output
使用huge模型
3327
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
