介绍

使用Meta AI的SAM，并添加了一个基本界面来标记图像，并保存COCO格式的掩码。

源码

https://github.com/anuragxel/salt

安装

1. 安装SAM；
2. 创建conda虚拟环境，使用conda env create -f environment.yaml;
3. 安装coco-viewer来快速可视化标注结果。

使用方法

1、将图片放入到<dataset_name>/images/*并且创建空目录<dataset_name>/embeddings
标签会自动保存在<dataset_name>/annotations.json
2、运行helpers脚本

• 运行extract_embeddings.py来提取图像的中间特征
• 运行generate_onnx.py来生成*.onnx文件，保存在models中。

3、 运行segment_anything_annotator.py，给相关的参数，包括<