手机:16601807362,可咨询购买自动驾驶开发套件、自动驾驶教学平台及解决方案、线控底盘、Mobileye相机、德尔福和博世4D雷达、激光雷达。
今天看啥  ›  专栏  ›  人工智能AI大模型与汽车自动驾驶

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

人工智能AI大模型与汽车自动驾驶  · 公众号  ·  · 2024-10-15 10:32

文章预览

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标准格 ………………………………

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