ISAT(Image Segmentation Annotation Tool)结合了Meta推出的Segment Anything Model(SAM),是一款功能强大且灵活的图像分割和打标签工具。
源码: https://github.com/yatengLG/ISAT_with_segment_anything/tree/master
官方中文安装及使用教程:https://isat-samzh.readthedocs.io/zh-cn/latest/
一、安装
1、3种安装方式
- 源码运行
- 通过pip安装
- 通过exe安装
2、源码安装步骤
系统及软件:win10+Anaconda
1、去官网下载源码,或者使用git拉取
git clone https://github.com/yatengLG/ISAT_with_segment_anything.git
2、准备环境
创建虚拟环境
conda create -n isat_env python=3.8
conda activate isat_env
安装pytorch
通过pip安装的pytorch默认是cpu版本,gpu版本的pytorch需去 pytorch 官网手动安装。
切换到ISAT_with_segment_anything项目文件夹下,安装需要的库:
cd ISAT_with_segment_anything
pip install -r requirements.txt
3、运行软件
python main.py
运行后打开的界面如下:
二、使用
1、下载预训练模型
2种下载模型的方式:
- 从网上下载模型,保存到以下目录
ISAT_with_segment_anything/ISAT/checkpoints
- 软件提供了模型管理界面。通过[菜单栏]-[SAM]-[模型管理]打开界面,可以下载、删除模型
2、常用设置
2.1 中英文切换、查看快捷键
2.2 自定义类
2.3 实例分割标注步骤
1、选择要标注图片所在的文件夹,选择标签要存放的文件夹
2、选择标注要使用的模型
注意:如果没有下载或者不选择模型,上图👀图标是灰色的,无法使用SAM标注功能
3、设置自动保存
如果不设置自动保存,每次切换到下一张图片时,会提示你:是否要保存更改
4、打标签
- 先选择类别,键盘上按下Q键调用模型准备标记
- 鼠标左键在目标物体上点击,自动框选,可多次点击鼠标左键选择想要区域,有不想要的区域时按鼠标右键,按下键盘E键保存
- 当一张图像上有多个物体时,重复上面的步骤 ;右侧标注信息可以删除、编辑等
- 一幅图标记好后按D键下一张,A键上一张,撤回按Z键
5、转换成yolo格式