【学习笔记】基于Segament-anything衍生标注工具使用

本文章主要记录工具学习使用

项目地址

https://github.com/haochenheheda/segment-anything-annotator

1.工具启动

python annotator.py --app_resolution 1000,1600 --model_type vit_b --keep_input_size True --max_size 720

–model_type: vit_b, vit_l, vit_h（模型选择，本项目使用vit_b,）
–keep_input_size: True: 保留 SAM 的原始图像大小；False：将输入图像调整为–max_size（节省 GPU 内存）

2.使用

加载模型
工具启动后点击 Load SAM加载SAM模型，加载成功终端会出现vit_b model already exists as 'vit_b.pth'. Skipping download.
打开数据集
点击 Category File 加载数据集标签文件，点击Image/Save Directory 选择图片/标注文件保存地址。
开始标注
点击Manual Polygon 可对图片进行手动标注，左键进行点点标注，右键按住可直接进行绘画标注。
点击Point/Box Prompt 可以选择点标注建议，框标注建议：
Point Prompt 在目标中点击一下，即可生成四种标注建议(快捷选择键"1,2,3,4)
Box Prompt 左键一下选择矩形起点，再次点击选择矩形终点，工具会对矩形内目标进行生成标注建议。
Accept/Reject 接受标注建议转换成标注框/拒绝标注建议(快捷键a,r)
编辑模式
Edit Polygons 进入编辑模式可对完成标注的标注框进行修改。
Edit Label 编辑模式下选中标注框可对标签进行修改
撤回/删除/复制/简化
Undo 可进行上一步的撤回
Delete Polygons 编辑模式下可选中标注框可删除
Duplicate Polygons 编辑模式下选中标注框可复制标注框
Reduce Points 编辑模式下选中标注框可减少标注点(对辅助标注的标注框点较多不方便进行修改时，可通过减少标注点重新修改)
保存
标注完成后需点击Save 进行手动保存，工具暂无自动保存功能。

3.结尾

此标注工具可针对某些图片相同目标较多的图片时，可提升较高的效率。或遇到标注比较麻烦的目标也可优先选择此工具。现在还是有比较多的功能与快捷键不完善。
此项目还有标注视频的工具，运行程序为annotator_video.py。我因为网络原因下载不了预训练模型就放弃使用，感兴趣的朋友可以尝试一下，使用方法都大同小异。