目录
安装文档(中文):欢迎查阅ISAT-SAM的文档! — ISAT-SAM 文档 (isat-samzh.readthedocs.io)
一、简介
ISAT-SAM 是一款基于 Segment Anything Model 的交互式半自动图像分割标注工具
标注文件储存在类似MSCOCO格式的json文件里:
-
info:
-
description:始终为 ‘ISAT’
-
folder:存储图像的目录
-
name:图像文件的名称
-
width, height, depth: 图片的尺寸和深度; RGB图为3通道深度
-
note:注释, 与图像相关的任何附加说明的可选字段
-
-
objects:
-
category: 标注的种类
-
group: 分组, 从1开始计算
-
segmentation: 分割的坐标点 [[x1, y1], [x2, y2], …, [xn, yn]]
-
area:对象覆盖的像素区域
-
layer: 图层, 从1.0开始,每层 + 1
-
bbox: 边框格式 [x_min, y_min, x_max, y_max]
-
iscrowd:布尔值,指示对象是否为群体
-
二、安装
源码运行
conda create -n ISAT python==3.8 -y
conda activate ISAT
获取ISAT_with_segment_anything源码并安装依赖windows系统下,通过pip安装的pytorch默认是cpu版本,gpu版本的pytorch需去 pytorch 官网手动安装。
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
这个具体版本需要根据你的cuda版本去确定。
git clone https://github.com/yatengLG/ISAT_with_segment_anything.git
cd ISAT_with_segment_anything
pip install -r requirements.txt
下载Segment anything预训练模型: 保存到以下目录``ISAT_with_segment_anything/ISAT/checkpoints``
软件提供了模型管理界面。通过[菜单栏]-[SAM]-[模型管理]打开界面(现已添加模型国内下载链接,点击下载按钮可直接进行下载)
运行软件
python main.py
三、使用记录
参考:功能 — ISAT-SAM 文档 (isat-samzh.readthedocs.io)
conda activate ISAT
cd E:\ISAT_with_segment_anything
python main.py
选择你的模型
创建文件夹
将对应内容存储
基本说明
左键:选择感兴趣的区域
右键:取消选择(不感兴趣区域)
E:结束标注,选择类别
拖拽多边形顶点,精细化调整标注。
通过目标图层高低,调整目标之间遮挡关系(多目标之间存在重叠区域时)
常用快捷键
q:开启模型标注模式
c: 开始正常标注模式
e: 结束标注,选择类别
s: 保存
d: 下一张图
a: 上一张图
注意事项
- 仅支持打开文件下的图片,不支持递归子文件夹内图片
- 如果多个文件夹分批次标注,需要关闭程序,重新运行。如果当前文件夹标注完成,不关闭程序,打开(加载)新的文件,若两个文件夹内图片名称有重复,可能出现混乱
- 对于不好标注图片需要用常规标注模式