开源一周内,Github 24k star,使用Segment Anything(SAM)模型进行自动标注


添加微信:kkcoder,备注:CV、NLP,拉你入群,一起学习。
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 <dataset_path> --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 <dataset_path> --categories cat,dog



# cd到项目1的主目录下
python cocoviewer.py -i <dataset_path> -a <dataset_path>\annotations.json

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






原文地址
https://blog.csdn.net/weixin_45977690/article/details/130088039
最后
添加微信:kkcoder,备注:CV、NLP,拉你入群,一起学习。

好了,今天的内容先这样,继续想看解决什么问题,评论区留言~
最近我们花了几个月整理的一个内容,可meeting的idea !!
绝对绝对不可错过的一个内容~ 

都到这里了,记得点赞哦~
到顶部