hughie

工具分享#

​ FastSAM是一个 CNN Segment Anything 模型，仅使用 SA-1B 数据集的 2％进行训练。FastSAM 在 50 倍的运行速度下实现了与 SAM 方法相当的性能。

特点#

官方介绍比SAM快 50 倍，并且打到相同性能效果，模型基于 YoloV8 结构，训练成本也比 SAM 的低很多。

部署#

• 安装项目

  Copy

  git clone https://github.com/CASIA-IVA-Lab/FastSAM.git && cd FastSAM
  pip install git+https://github.com/openai/CLIP.git
  
  conda create -n FastSAM python=3.9
  conda activate FastSAM
  
  pip install -r requirements.txt
  

• 使用

  Copy

  # Everything模式
  python Inference.py --model_path ./weights/FastSAM.pt --img_path ./images/dogs.jpg
  
  # 文本提示分割，使用CLIP辅助
  python Inference.py --model_path ./weights/FastSAM.pt --img_path ./images/dogs.jpg  --text_prompt "the yellow dog"
  
  # 框选模式
  python Inference.py --model_path ./weights/FastSAM.pt --img_path ./images/dogs.jpg --box_prompt "[570,200,230,400]"
  
  # 点击模式
  python Inference.py --model_path ./weights/FastSAM.pt --img_path ./images/dogs.jpg  --point_prompt "[[520,360],[620,300]]" --point_label "[1,0]"
  

平台#

官方 demo

声明#

本文仅作为工具分享记录。

本文与HBlog保持同步。