专栏名称: 小白学视觉
本公众号主要介绍机器视觉基础知识和新闻,以及在学习机器视觉时遇到的各种纠结和坑的心路历程。
今天看啥  ›  专栏  ›  小白学视觉

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

小白学视觉  · 公众号  ·  · 2024-09-10 10:08

文章预览

点击上方 “ 小白学视觉 ”,选择加" 星标 "或“ 置顶 ” 重磅干货,第一时间送达 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 \ann ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览