使用Segment Anything(SAM)模型进行自动标注

Meta开源了一个图像分割模型【SegmentAnything Model】，简称 SAM 模型，号称分割一切，在短短开源的一周内，截止今天Github已经24k的star了！

1.下载项目

项目1：https://github.com/zhouayi/SAM-Tool

项目2：https://github.com/facebookresearch/segment-anything

下载SAM模型：

https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth

# cd到项目2的主目录下

python helpers\extract_embeddings.py --checkpoint-path sam_vit_h_4b8939.pth --dataset-folder --device cpu

# cd到项目2的主目录下

python helpers\generate_onnx.py --checkpoint-path sam_vit_h_4b8939.pth --onnx-model-path ./sam_onnx.onnx --orig-im-size 1080 1920

# cd到项目1的主目录下

python segment_anything_annotator.py --onnx-model-path sam_onnx.onnx --dataset-path --categories cat,dog

# cd到项目1的主目录下

python cocoviewer.py -i -a \annotations.json

6.将保存的json格式转换为voc格式

该工具保存的标注文件格式为COCO标准格式，如有需求可自行编写标注文件格式转换脚本，下面提供转换为VOC格式的脚本例子

原文地址

https://blog.csdn.net/weixin_45977690/article/details/130088039

下载1：OpenCV-Contrib扩展模块中文版教程
在「小白学视觉」公众号后台回复：扩展模块中文教程，即可下载全网第一份OpenCV扩展模块教程中文版，涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。

下载2：Python视觉实战项目52讲
在「小白学视觉」公众号后台回复：Python视觉实战项目，即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目，助力快速学校计算机视觉。

下载3：OpenCV实战项目20讲
在「小白学视觉」公众号后台回复：OpenCV实战项目20讲，即可下载含有20个基于OpenCV实现20个实战项目，实现OpenCV学习进阶。

交流群