使用sam进行零样本、零学习的分割实践

参照：利用SAM实现自动标注_sam标注-CSDN博客，以及SAM（分割一切模型）的简单调用_sam使用-CSDN博客

sam简介：

Segment Anything Model（SAM）是Meta公司于2023年发布的一种AI模型，它打破了现有的分割技术，通过解决任意输入图像的分割问题，开启了发现世界中对象和区域可能性的新时代。 SAM对人类数字世界和整个视觉应用生态系统有着深远的影响，推动软件与用户的互动方式达到了新的高度。

Segment Anything Model的主要特点包括：

**1. 卓越的准确性：**在最为复杂的COCO panoptic基准上，SAM以超过56%的mask quality领先于其他方法。模型的准确性与之前的最佳方法相比，提高了30%，并且比亚军模型高出10个百分点。

**2. 即插即用：**模型以一种即插即用的方式提供，不需要重新训练或微调，同时能够泛化到之前未见过的图像和任务。

**3. 遮罩调优能力：**用户可以使用单个点击（称为点提示）来校正或微调模型的初始分割结果。

**4. 多模式提示：**模型能够采用来自多种模式的输入，包括复选框、排除框和标记点，用于优化分割。

**5. 灵活性：**支持处理各种任务，从细粒度的边界到粗糙的分割。

**6. 速度：**仅需约0.3秒/帧的速度，在快速和准确之间取得了平衡，与现有的交互式分割方法相比，显著减少了时间需求。

**7. 强大的泛化能力：**SAM 在不同的数据集和应用场景下都表现出了良好的泛化能力，不仅可以处理各种类型的图像，对于一些它未见过或相对模糊的场景，同样能实现较好的图像分割效果。

总的来说，SAM提供了一种高效的分割解决方案，具有高度的灵活性和准确性，同时不需要像传统的分割方法那样进行大量的微调和训练。

本文通过一个简单的图片分割实践来体验一下sam在非标图片的零样本、零训练下的分割能力。

一、前期准备

1、下载或克隆项目，打开下面的网址，可以克隆或者下载zip文件，这里选择了下载zip文件：GitCode - 全球开发者的开源社区,开源代码托管平台

得到了文件：segment-anything-main.zip

在这个网页中，还可以看到环境的要求：

2、按照上面的要求在conda虚拟环境下安装pytorch和torchvision。这里要注意，虽然SAM没有直接对CUDA版本提出要求，但是pytorch和CUDA的版本是紧密相关的。

选择了：

pytorch=1.7.1

torchvision=0.8.0

python=3.8

cuda=11.0

cudnn=8.0.5

具体的版本选择方法见：人工智能学习用的电脑安装cuda、torch、conda等软件，版本的选择以及多版本切换_cuda版本要求-CSDN博客

如果使用在线的方式：Previous PyTorch Versions | PyTorch

根据选定的版本下载和安装相关的torch、cuda、conda等工具和软件。

3、安装必要的依赖包

复制代码

pip install opencv-python pycocotools matplotlib onnxruntime onnx

4、下载SAM-TOOL：GitCode - 全球开发者的开源社区,开源代码托管平台，得到文件SAM-Tool-main.zip。

5、下载模型：https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth

得到文件：sam_vit_h_4b8939.pth

6、安装：将segment-anything-main.zip解压缩，进入SAM所在的文件夹目录后，在目录下鼠标右键-->"打开于终端",然后：

复制代码

pip install -e .

二、牛刀小试

1、安装完成后，在segment-anything-main目录下新建目录imgs用于存放图片，新建models目录用以存放模型权重，并新建demo.py，作为测试的脚本。

2、将前面下载得到的阶段权重模型sam_vit_h_4b8939.pth，复制到models目录。

3、将测试用的图片复制到imgs目录。图片是一个工业产品的微观图，特征如下：

图片的大小为1200*850像素，将(330,390)与(480,550)框定的范围作为采样的样本。

4、demo.py的代码内容：

复制代码

import torch
from segment_anything import SamPredictor, sam_model_registry
import cv2
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image

# 加载模型
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
sam_checkpoint = "models/sam_vit_h_4b8939.pth"  # 这里填写模型权重路径
model_type = "vit_h"   # 模型类型，可选vit_h, vit_l, vit_b
sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)   # 加载模型
sam.to(device=device)   # 将模型加载到GPU或CPU

# 初始化SAM预测器
predictor = SamPredictor(sam)

# 读取图片
image_path = "/home/dy/program/segment-anything-main/imgs/IMG_PP.jpg"  # 图片路径
image = cv2.imread(image_path)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)   # 将图片转换为RGB格式

# 设置输入图片
predictor.set_image(image)

# 指定一些提示点，指导SAM识别特定对象
input_point = np.array([[330, 390], [480, 550]])  # 将列表转换为NumPy数组
input_label = np.array([1, 1])  # 也将标签转换为NumPy数组

# 获得掩膜
masks, scores, logits = predictor.predict(   # 返回掩膜、置信度、类别概率
    point_coords=input_point,   # 指定提示点
    point_labels=input_label,   # 指定提示点对应的标签
    multimask_output=True,     # 是否输出多类别掩膜
)

print("置信度：", scores,
      "类别概率:", logits)   # 显示置信度, 类别概率


def show_masks(masks):   # 显示掩膜
    for i, mask in enumerate(masks):
        mask_image = (mask * 255).astype("uint8")    # 将掩膜转换为8位无符号整数
        mask = Image.fromarray(mask_image)     # 将掩膜转换为PIL图像
        mask = mask.convert("L")    # 将掩膜转换为灰度图像
        plt.imshow(mask, cmap='jet', alpha=0.5)    # 显示掩膜


# 显示结果
plt.figure(figsize=(10, 10))     #  设置画布大小
plt.imshow(image)     # 显示原图
show_masks(masks)    # 显示掩膜
plt.axis('off')      #  隐藏坐标轴
plt.show()    # 显示图像

5、运行脚本，输出结果：

看得出，针对一个零样本、零训练的非标图片，sam的表现还是非常优异的。